Home
last modified time | relevance | path

Searched refs:indexing_maps (Results 1 – 25 of 66) sorted by relevance

123

/external/llvm-project/mlir/test/Dialect/Linalg/
Dfold-unit-trip-loops.mlir10 indexing_maps = #accesses,
27 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP1]]]
36 indexing_maps = #access,
52 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP0]]]
61 indexing_maps = #access,
78 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP0]]]
89 indexing_maps = #accesses,
106 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP1]]]
Dfusion-tensor.mlir9 …%0 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
16 // CHECK-SAME: indexing_maps = {{\[}}[[$MAP0]], [[$MAP0]], [[$MAP0]], [[$MAP0]]{{\]}}
17 …%2 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
44 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle…
51 // CHECK-SAME: indexing_maps = {{\[}}[[$MAP0]], [[$MAP1]], [[$MAP0]], [[$MAP0]]{{\]}}
52 …%2 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
71 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle…
78 // CHECK-SAME: indexing_maps = {{\[}}[[$MAP1]], [[$MAP0]], [[$MAP0]], [[$MAP0]]{{\]}}
79 …%2 = linalg.generic {indexing_maps = [#map1, #map0, #map0], iterator_types = ["parallel", "paralle…
99 %0 = linalg.generic {indexing_maps = [#map2, #map2, #map2], iterator_types = ["parallel"]}
[all …]
Dreshape_linearization_fusion.mlir17 indexing_maps = [#map0, #map0, #map0],
29 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP1]]]
44 indexing_maps = [#map0, #map0, #map0],
59 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP0]], #[[$MAP1]]]
70 indexing_maps = [#map0, #map0, #map0],
100 indexing_maps = [#map0, #map0],
114 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP1]]]
126 indexing_maps = [#map0, #map0],
143 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP1]]]
157 …%1 = linalg.generic {indexing_maps = [#map2, #map3], iterator_types = ["parallel", "parallel", "pa…
[all …]
Dsparse_invalid.mlir4 indexing_maps = [
29 indexing_maps = [
56 indexing_maps = [
82 indexing_maps = [
102 indexing_maps = [
128 indexing_maps = [
150 indexing_maps = [
175 indexing_maps = [
200 indexing_maps = [
225 indexing_maps = [
Dinvalid.mlir46 indexing_maps = [ affine_map<() -> (0)> ],
56 indexing_maps = [ affine_map<() -> ()> ],
69 indexing_maps = [ affine_map<() -> (0)> ],
82 indexing_maps = [ affine_map<() -> (0, 0)> ],
95 indexing_maps = [ affine_map<(i) -> (i)> ],
109 indexing_maps = [
131 indexing_maps = [ affine_map<() -> (0)> ],
148 indexing_maps = [ affine_map<() -> (0)> ],
160 indexing_maps = [ affine_map<() -> (0)> ],
173 indexing_maps = [ affine_map<() -> (0)> ],
[all …]
Ddrop-unit-extent-dims.mlir10 indexing_maps = #accesses,
33 // CHECK-SAME: indexing_maps = [#[[$MAP2]], #[[$MAP3]]]
46 indexing_maps = #accesses,
84 indexing_maps = #access,
101 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP0]]]
110 indexing_maps = #access,
141 indexing_maps = #accesses,
158 // CHECK-SAME: indexing_maps = [#[[$MAP1]], #[[$MAP1]]]
170 indexing_maps = #accesses,
195 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]]
[all …]
Dcanonicalize-duplicate-inputs.mlir13 …%0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel"]} ins(%arg0,…
34 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle…
57 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0, #map0], iterator_types = ["parallel", "…
77 …%0 = linalg.generic {indexing_maps = [#map, #map, #map, #map, #map], iterator_types = ["parallel"]…
98 …%0 = linalg.indexed_generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel"]} in…
Dreshape_fusion.mlir14 indexing_maps = [#map0, #map1, #map1],
36 // CHECK-SAME: indexing_maps = [#[[MAP3]], #[[MAP4]], #[[MAP4]]]
52 indexing_maps = [#map0, #map0, #map0],
78 // CHECK-SAME: indexing_maps = [#[[MAP2]], #[[MAP2]], #[[MAP2]]]
90 indexing_maps = [affine_map<(d0, d1, d2) -> (d1, d0, d2)>,
124 // CHECK-SAME: indexing_maps = [#[[MAP5]], #[[MAP6]], #[[MAP7]]]
139 indexing_maps = [#map0, #map0, #map0],
160 // CHECK-SAME: indexing_maps = [#[[MAP2]], #[[MAP2]]]
171 {indexing_maps = [affine_map<(d0) -> ()>, affine_map<(d0) -> (d0)>],
189 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]]]
[all …]
Dfusion-indexed-generic.mlir6 indexing_maps = [#id_2d, #id_2d, #id_2d],
35 indexing_maps = [#id_2d, #id_2d],
68 indexing_maps = [#id_2d, #id_2d, #id_2d],
102 indexing_maps = [#id_2d, #id_2d],
130 indexing_maps = [#id_2d, #id_2d, #id_2d],
168 indexing_maps = [#id_2d, #id_2d],
Dbufferize.mlir16 // CHECK: linalg.generic {indexing_maps = [#map, #map], iterator_types = ["parallel"]}
27 indexing_maps = [#map0, #map0],
51 indexing_maps = [#map0, #map0, #map0],
75 indexing_maps = [#map0, #map0, #map0],
105 // CHECK: linalg.generic {indexing_maps = [#map0, #map0, #map1]
111 indexing_maps = [#map_2d, #map_2d, #map_2d_inv],
129 indexing_maps = #accesses,
Dparallel-loops.mlir8 indexing_maps = [#map0, #map0, #map0],
38 indexing_maps = #accesses
71 indexing_maps = #accesses
Dconvert-elementwise-to-linalg.mlir7 …// CHECK: %{{.*}} = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} ins(%…
22 // CHECK: linalg.generic{{.*}}indexing_maps = [#map, #map, #map], iterator_types = ["parallel"]
Dsparse_parallel.mlir13 indexing_maps = [
62 indexing_maps = [
111 indexing_maps = [
Dgeneralize-named-ops.mlir18 // CHECK-SAME: indexing_maps = [#[[FILTER_MAP]], #[[INPUT_MAP]], #[[OUTPUT_MAP]]]
46 // CHECK-SAME: indexing_maps = [#[[A_MAP]], #[[B_MAP]], #[[C_MAP]]]
Droundtrip.mlir296 indexing_maps = #accesses,
315 // CHECK-SAME: indexing_maps = [#{{[0-9a-z]*}}, #{{[0-9a-z]*}}],
336 // CHECK-SAME: indexing_maps = [#{{.*}}, #{{.*}}], iterator_types = ["parallel", "parallel", "…
346 linalg.generic {indexing_maps = [#map0],
369 indexing_maps = #accesses,
388 // CHECK-SAME: indexing_maps = [#{{.*}}, #{{.*}}], iterator_types = ["parallel", "parallel", "…
404 indexing_maps = #accesses,
423 // CHECK-SAME: indexing_maps = [#{{.*}}, #{{.*}}], iterator_types = ["parallel", "parallel", "…
438 indexing_maps = #broadcast_access,
474 indexing_maps = #accesses,
[all …]
Dinlining.mlir12 indexing_maps = #accesses,
/external/llvm-project/mlir/integration_test/Dialect/Vector/CPU/
Dtest-contraction.mlir12 indexing_maps = #dotp_accesses,
22 indexing_maps = #matvec_accesses,
32 indexing_maps = #mattransvec_accesses,
42 indexing_maps = #matmat_accesses,
52 indexing_maps = #mattransmat_accesses,
62 indexing_maps = #matmattrans_accesses,
72 indexing_maps = #mattransmattrans_accesses,
82 indexing_maps = #matmat_then_trans_accesses,
92 indexing_maps = #contract2d_accesses,
102 indexing_maps = #contract2d_alt_accesses,
[all …]
/external/llvm-project/mlir/test/Conversion/LinalgToSPIRV/
Dlinalg-to-spirv.mlir9 indexing_maps = [
67 indexing_maps = [
96 indexing_maps = [
127 indexing_maps = [
/external/llvm-project/mlir/test/Dialect/Vector/
Dvector-transforms.mlir74 indexing_maps = #contraction_accesses0,
96 // CHECK-NEXT: %[[R1S00:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
102 // CHECK-NEXT: %[[R2S00:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
108 // CHECK-NEXT: %[[R3S00:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
115 // CHECK-NEXT: %[[R1S02:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
119 // CHECK-NEXT: %[[R2S02:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
123 // CHECK-NEXT: %[[R3S02:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
130 // CHECK-NEXT: %[[R1S20:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_typ…
134 // CHECK-NEXT: %[[R2S20:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_ty…
138 // CHECK-NEXT: %[[R3S20:.*]] = vector.contract {indexing_maps = [#map0, #map1, #map2], iterator_ty…
[all …]
Dvector-contract-matvec-transforms.mlir9 indexing_maps = #matvec_accesses,
19 indexing_maps = #mattransvec_accesses,
29 indexing_maps = #vecmat_accesses,
39 indexing_maps = #vecmattrans_accesses,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo-fuse-linalg.mlir6 #pointwise_2d_trait = {indexing_maps = [#map0, #map0, #map0],
67 indexing_maps = [affine_map<(d0, d1) -> (d0)>,
77 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
89 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
139 #pointwise_4d_trait = {indexing_maps = [#map0, #map0, #map0],
199 #pointwise_2d_trait = {indexing_maps = [#map0, #map0, #map0],
262 linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>,
314 linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>,
384 linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>,
Dhlo-legalize-to-linalg.mlir393 // CHECK: linalg.generic {{{.*}}indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
409 // CHECK: linalg.generic {{{.*}}indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
425 // CHECK: linalg.generic {{{.*}}indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
442 // CHECK: linalg.generic {{{.*}}indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
458 // CHECK: linalg.generic {{{.*}}indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
472 // CHECK: linalg.generic {{{.*}}indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
596 // CHECK-SAME: indexing_maps = [#[[MAP]], #[[MAP]], #[[MAP]]]
804 // CHECK-SAME: indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
816 // CHECK-SAME: indexing_maps = [#[[RESULT_MAP]]]
950 // CHECK-SAME: indexing_maps = [#[[OPERAND_MAP]], #[[RESULT_MAP]]]
[all …]
/external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/
DLinalgToSPIRV.cpp91 if (genericOp.indexing_maps().getValue().size() != 2) in matchAsPerformingReduction()
96 auto inputMap = genericOp.indexing_maps().getValue()[0].cast<AffineMapAttr>(); in matchAsPerformingReduction()
98 genericOp.indexing_maps().getValue()[1].cast<AffineMapAttr>(); in matchAsPerformingReduction()
/external/llvm-project/mlir/test/mlir-linalg-ods-gen/
Dtest-linalg-ods-gen.tc13 // IMPL: ArrayAttr Test1Op::indexing_maps() {
38 // IMPL: ArrayAttr Test2Op::indexing_maps() {
63 // IMPL: ArrayAttr Test3Op::indexing_maps() {
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/
Dbuffer_reuse.mlir48 indexing_maps = [affine_map<(i) -> (i)>, affine_map<(i) -> (i)>],
73 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>],
94 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>],
111 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>],
124 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>],
363 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>],
435 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>],

123