cerebras.modelzoo.data.common.tensor_spec#

Wrapper class used to process TensorSpecs using a custom yaml tag.

Functions

tensor_spec_constructor

Constructor used to register TensorSpec in the yaml loader.

Classes

TensorSpec

Wrapper class used to wrap the leaf nodes in SyntheticDataProcessor's input.