Home
last modified time | relevance | path

Searched full:shape_map (Results 1 – 7 of 7) sorted by relevance

/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml20 shape_map: affine_map<() -> ()>
25 shape_map: affine_map<() -> ()>
61 shape_map: affine_map<() -> ()>
66 shape_map: affine_map<() -> ()>
110 shape_map: affine_map<() -> ()>
115 shape_map: affine_map<() -> ()>
145 shape_map: affine_map<() -> ()>
150 shape_map: affine_map<() -> ()>
180 shape_map: affine_map<() -> ()>
185 shape_map
[all...]
/llvm-project/mlir/test/python/dialects/linalg/opdsl/
H A Dshape_maps_iteration.py10 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s0, s1)>
11 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s1, s2)>
12 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s0, s2)>
37 # CHECK: shape_map: affine_map<()[s0] -> (s0)>
38 # CHECK: shape_map: affine_map<()[s0] -> (s0)>
39 # CHECK: shape_map: affine_map<()[s0] -> ()>
55 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s0)>
56 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s1)>
57 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s2)>
H A Darguments.py12 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s0, s1)>
16 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s1, s2)>
20 # CHECK: shape_map: affine_map<()[s0, s1, s2] -> (s0, s2)>
47 # CHECK-NOT: shape_map:
60 # CHECK: shape_map: affine_map<()[s0, s1, s2, s3, s4, s5] -> (s0, s1)>
64 # CHECK: shape_map: affine_map<()[s0, s1, s2, s3, s4, s5] -> (s2, s3)>
/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dconfig.py49 shape_map: Optional[_ir.AffineMap] = None,
53 self.shape_map = shape_map # type: Optional[_ir.AffineMap]
73 if self.shape_map:
74 self_dict["shape_map"] = _serialize_affine_map(self.shape_map)
86 f"shape_map={self.shape_map}, "
228 if operand_config.shape_map:
229 operand_config.shape_map = self._normalize_affine_map(
230 operand_config.shape_map, with_dims=False
357 operand_def, shape_map=affine_map
H A Demitter.py598 if operand_config.shape_map:
/llvm-project/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml27 shape_map: affine_map<()[s0, s1] -> (s0, s1)>
130 shape_map: affine_map<()[s0, s1, s2, s3] -> (s0, s1)>
135 shape_map: affine_map<()[s0, s1, s2, s3] -> (s0, s1)>
221 shape_map: affine_map<() -> ()>
271 shape_map: affine_map<()[s0, s1] -> (s0, s1)>
358 shape_map: affine_map<() -> ()>
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp180 /// - `shape_map`: An optional AffineMap from all op symbols to the shape of
181 /// the argument. Only tensor arguments have a `shape_map`. Each shape must
198 io.mapOptional("shape_map", info.shapeMap); in mapping()