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.

previous

cerebras.modelzoo.data.common.restartable_dataloader.RestartableDataLoader

next

cerebras.modelzoo.data.common.tensor_spec.tensor_spec_constructor