ShardedDim¶
- class onnx_ir.ShardedDim(axis, simple_shardings=())¶
The sharding of a single tensor axis.
Added in version 1.0.0.
- Parameters:
axis (int)
simple_shardings (tuple[SimpleShardedDim, ...])
- simple_shardings: tuple[SimpleShardedDim, ...] = ()¶