cerebras.modelzoo.data.common.tensor_spec#
Wrapper class used to process TensorSpecs using a custom yaml tag.
Functions
Constructor used to register TensorSpec in the yaml loader. |
Classes
Wrapper class used to wrap the leaf nodes in SyntheticDataProcessor's input. |