SimpleShardedDim

class onnx_ir.SimpleShardedDim(dim=<factory>, num_shards=0)

How one axis is divided into shards (N blocks split into num_shards).

Added in version 1.0.0.

Parameters:

• dim (int | SymbolicDim)

• num_shards (int)

num_shards: int = 0

dim: int | SymbolicDim