Skip to main content

ShapeDomain

One input's admitted shapes: its name, its element type, one DimDomain per dimension, and the axis a benchmark counts items along. batch_axis is a guess the report echoes back: the first dimension named batch, batch_size, N or B, else the first dynamic dimension, else -1 (one item per run); BenchOptions(batch_axis=...) overrides it.

batch_axis (property)​

The guessed batch axis, or -1.

dims (property)​

One DimDomain per dimension.

dtype (property)​

The input's element type.

name (property)​

The input's name.

__init__​

__init____init__(self) -> None

init(self) -> None

admits​

admitsadmits(self, dims: collections.abc.Sequence[int]) -> bool

admits(self, dims: collections.abc.Sequence[int]) -> bool

admits(dims) -> bool

True when dims has the input's rank and every extent is admitted by its DimDomain.