cerebras.modelzoo.data.common.tensor_spec.tensor_spec_constructor#

cerebras.modelzoo.data.common.tensor_spec.tensor_spec_constructor(loader: yaml.loader.SafeLoader, node: yaml.nodes.MappingNode)[source]#

Constructor used to register TensorSpec in the yaml loader.

previous

cerebras.modelzoo.data.common.tensor_spec

next

cerebras.modelzoo.data.common.tensor_spec.TensorSpec