cerebras.modelzoo.common.input_utils.PaddingSample

class cerebras.modelzoo.common.input_utils.PaddingSample(shape, dtype, require_grad=False)

Bases: torch.Tensor

Methods

Attributes

is_padding

previous

cerebras.modelzoo.common.input_utils.validate_streaming_and_micro_batch_size

next

cerebras.modelzoo.common.mlops