Searched refs:DimDef (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | comprehension.py | 39 if isinstance(dim_def, DimDef):
117 def _compute_reduce_dims(self, rhs: TensorExpression) -> Set[DimDef]:
230 def __init__(self, dim: DimDef): argument
359 binary_attr: Optional["BinaryFnAttrDef"], *reduce_dims: DimDef): argument
386 def __getitem__(self, reduce_dims: Tuple[DimDef]) -> ReduceFnUse: argument
428 index_dims: Optional[Sequence[DimDef]] = None, argument
491 index_dims: Optional[Sequence[DimDef]] = None, argument
496 if index_dims and any(not isinstance(dim, DimDef) for dim in index_dims):
618 def __getitem__(self, reduce_dims: Tuple[DimDef]) -> ReduceFnUse: argument
663 def all_reduction_dims(self) -> Set[Tuple[DimDef, ...]]:
|
| H A D | dsl.py | 152 def domain(*dimensions: DimDef): argument
153 if any(not isinstance(d, DimDef) for d in dimensions):
|
| H A D | affine.py | 227 class DimDef(AffineExprDef): class
306 D = DimDef.create_expando()
|
| H A D | config.py | 119 domain: Sequence[DimDef], argument
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 846 index_vars: Tuple[IndexVar, ...]) -> Tuple[lang.DimDef, ...]:
|