Searched defs:DimDef (Results 1 – 4 of 4) sorted by relevance
| /llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | comprehension.py | 35 def collect_dim_uses(self, uses: Set["DimDef"]): argument
242 def __init__(self, dim: DimDef): argument
413 __init__( self, binary_fn: Optional[BinaryFnType], binary_attr: Optional["BinaryFnAttrDef"], *reduce_dims: DimDef, ) global() argument
440 __getitem__(self, reduce_dims: Tuple[DimDef]) global() argument
484 __init__( self, kind: OperandKind, type_var: Optional[TypeVar] = None, size_exprs: Optional[Sequence[AffineExprDef]] = None, index_dims: Optional[Sequence[DimDef]] = None, default_indices: Optional[Sequence[int]] = None, default_fn: Optional[str] = None, ) global() argument
554 __init__( self, type_var: TypeVar, *shape: AffineExprDef, index_dims: Optional[Sequence[DimDef]] = None, output: bool = False, ) global() argument
699 __getitem__(self, reduce_dims: Tuple[DimDef]) global() argument
726 __getitem__(self, reduce_dims: Tuple[DimDef]) global() argument
[all...] |
| H A D | dsl.py | 182 def domain(*dimensions: DimDef):
|
| H A D | config.py | 127 domain: Sequence[DimDef],
|
| H A D | affine.py | 230 class DimDef(AffineExprDef): class
|