cerebras.modelzoo.common.input_utils.PaddingSample#

class cerebras.modelzoo.common.input_utils.PaddingSample[source]#

Bases: torch.Tensor

Methods

Attributes

is_padding

static __new__(cls, shape, dtype, require_grad=False)[source]#

__call__(*args: Any, **kwargs: Any) → Any#: Call self as a function.

__init__(*args: Any, **kwargs: Any) → None#

previous

cerebras.modelzoo.common.input_utils.validate_streaming_and_micro_batch_size

next

cerebras.modelzoo.common.mlops