Searched defs:DimSpec (Results 1 – 1 of 1) sorted by relevance
45 class DimSpec: class76 def new(cls, dim: DimSpec, dim_size: int) -> DimSpec:102 def new(cls, dim: DimSpec, times: int) -> DimSpec:122 def new(cls, dims: Sequence[DimSpec]) -> DimSpec:149 def new(cls, dim: DimSpec, group_shape: Tuple[int, ...], idx: int) -> DimSpec:509 def collect_used_inputs(cmd: DimSpec) -> None:520 def get_in_dim_to_shard(cmd: DimSpec) -> Optional[InputDim]: