Home
last modified time | relevance | path

Searched full:index_attr_map (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dconfig.py50 index_attr_map: Optional[_ir.AffineMap] = None,
54 self.index_attr_map = index_attr_map # type: Optional[_ir.AffineMap]
75 if self.index_attr_map:
76 self_dict["index_attr_map"] = _serialize_affine_map(self.index_attr_map)
87 f"index_attr_map={self.index_attr_map}, "
232 if operand_config.index_attr_map:
233 operand_config.index_attr_map = self._normalize_affine_map(
234 operand_config.index_attr_map, with_dims=False
353 operand_def, index_attr_map=affine_map
H A Demitter.py91 results = index_attr.index_attr_map.results # type: AffineExprList
/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml2434 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6] -> (s2)>
2440 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6] -> (s4)>
2522 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6] -> (s3)>
2528 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6] -> (s5)>
2612 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10] ->
2620 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10] ->
2710 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10] ->
2718 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10] ->
2817 index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10] ->
2825 index_attr_map
[all...]
/llvm-project/mlir/test/python/dialects/linalg/opdsl/
H A Darguments.py67 # CHECK: index_attr_map: affine_map<()[s0, s1, s2, s3, s4, s5] -> (s4, s5)>
/llvm-project/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml139 index_attr_map: affine_map<()[s0, s1, s2, s3] -> (s2, s3)>
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp184 /// - `index_attr_map`: An optional AffineMap from all op symbols to the
187 /// arguments have an `index_attr_map`.
199 io.mapOptional("index_attr_map", info.indexAttrMap); in mapping()