Searched refs:TensorUse (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | comprehension.py | 43 if isinstance(expr, TensorUse):
56 if isinstance(expr, TensorUse):
92 class TensorUse(TensorExpression): class
503 def __getitem__(self, dims: Sequence[AffineExprDef]) -> TensorUse:
520 return TensorUse(self.operand_def, exprs)
649 def __init__(self, *bindings: Tuple[TensorUse, TensorExpression]): argument
|
| H A D | config.py | 33 def __init__(self, tensor_use: TensorUse, indexing_map: _ir.AffineMap): argument
344 def add_tensor_use(self, tensor_use: TensorUse): argument
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 562 dst_use = lang.TensorUse(dst_opnd, dst_dim_syms)
1664 return lang.TensorUse(expr_to_opnd[self], dims)
1740 return lang.TensorUse(expr_to_opnd[self], dims)
1800 return lang.TensorUse(expr_to_opnd[self], dims)
|