Home
last modified time | relevance | path

Searched refs:permutation_map (Results 1 – 25 of 31) sorted by relevance

12

/external/llvm-project/mlir/test/Conversion/VectorToROCDL/
Dvector-to-rocdl.mlir7 {permutation_map = affine_map<(d0) -> (d0)>} :
17 {permutation_map = affine_map<(d0) -> (d0)>} :
27 {permutation_map = affine_map<(d0) -> (d0)>} :
41 {permutation_map = affine_map<(d0) -> (d0)>} :
50 {permutation_map = affine_map<(d0) -> (d0)>} :
59 {permutation_map = affine_map<(d0) -> (d0)>} :
/external/llvm-project/mlir/test/mlir-rocm-runner/
Dvector-transferops.mlir10 {permutation_map = affine_map<(d0) -> (d0)>} :
17 {permutation_map = affine_map<(d0) -> (d0)>} :
32 {permutation_map = affine_map<(d0) -> (d0)>} :
38 {permutation_map = affine_map<(d0) -> (d0)>} :
/external/llvm-project/mlir/include/mlir/Interfaces/
DVectorInterfaces.td64 /*desc=*/"Return the `permutation_map` attribute name.",
69 /*defaultImplementation=*/ [{ return "permutation_map"; }]
112 /*methodName=*/"permutation_map",
114 /*methodBody=*/"return $_op.permutation_map();"
152 "return $_op.permutation_map().getNumResults();"
188 with `permutation_map = (d0, d1, d2, d3) -> (d2, d3)`.
/external/llvm-project/mlir/test/Dialect/Affine/SuperVectorize/
Dvectorize_outer_loop_transpose_2d.mlir28 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
49 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
52 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
54 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
Dvectorize_transpose_2d.mlir28 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
49 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
52 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
54 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
Dvectorize_outer_loop_2d.mlir16 … = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #[[map_proj…
Dvector_utils.mlir59 // VECNEST-NEXT: vector.transfer_write %{{.*}}, %{{.*}}[%{{.*}}, %{{.*}}] {permutation_map = #…
Dvectorize_2d.mlir126 …%[[A:.*]] = vector.transfer_read %{{.*}}[%[[I4]], %[[I3]]], %{{.*}} {permutation_map = #[[$map_pro…
127 …%[[B:.*]] = vector.transfer_read %{{.*}}[%[[I2]], %[[I4]]], %{{.*}} {permutation_map = #[[$map_pro…
Dvectorize_1d.mlir33 // CHECK-NEXT: {{.*}} = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}} {permutation_map = …
338 // CHECK: {{.*}} = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}} {permutation_map = #…
367 // CHECK-NEXT: {{.*}} = vector.transfer_read %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}} {permutation_map = …
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransferOpTransforms.cpp104 write.permutation_map() == write.permutation_map() && in deadStoreOp()
179 write.permutation_map() == read.permutation_map()) { in storeToLoadForwarding()
DVectorTransforms.cpp593 if (!isIdentitySuffix(readOp.permutation_map())) in unrollTransferReadOp()
616 readOp.permutation_map(), readOp.padding(), in unrollTransferReadOp()
637 if (!isIdentitySuffix(writeOp.permutation_map())) in unrollTransferWriteOp()
655 writeOp.permutation_map(), in unrollTransferWriteOp()
707 if (!isIdentitySuffix(xferReadOp.permutation_map())) in matchAndRewrite()
739 if (!isIdentitySuffix(xferWriteOp.permutation_map())) in matchAndRewrite()
772 xferWriteOp.permutation_map(), in matchAndRewrite()
2130 assert(xferOp.permutation_map().isMinorIdentity() && in createScopedInBoundsCond()
2157 if (!xferOp.permutation_map().isMinorIdentity()) in splitFullAndPartialTransferPrecondition()
2547 indices, read.permutation_map(), in matchAndRewrite()
[all …]
/external/llvm-project/mlir/integration_test/Dialect/Vector/CPU/
Dtest-transfer-write.mlir10 {permutation_map = affine_map<(d0) -> (d0)>, masked = [false]}
19 {permutation_map = affine_map<(d0) -> (d0)>}
28 {permutation_map = affine_map<(d0) -> (d0)>}
37 {permutation_map = affine_map<(d0) -> (d0)>}
Dtest-transfer-read.mlir9 {permutation_map = affine_map<(d0) -> (d0)>} :
18 {permutation_map = affine_map<(d0) -> (d0)>, masked = [false]} :
28 {permutation_map = affine_map<(d0) -> (d0)>} :
Dtest-transfer-to-loops.mlir57 …%1 = vector.transfer_read %0[%c1, %c1], %cst {permutation_map = #map0} : memref<?x?xf32>, vector<5…
67 …vector.transfer_write %1, %0[%c0, %c0] {permutation_map = #map0} : vector<5x5xf32>, memref<?x?xf32>
90 …%4 = vector.transfer_read %0[%c2, %c3], %cst {permutation_map = #map0} : memref<?x?xf32>, vector<5…
99 …%5 = vector.transfer_read %0[%c2, %c3], %cst {permutation_map = #map1} : memref<?x?xf32>, vector<5…
Dtest-transfer-read-2d.mlir9 {permutation_map = affine_map<(d0, d1) -> (d0, d1)>} :
19 {permutation_map = affine_map<(d0, d1) -> (d0, d1)>} :
/external/llvm-project/mlir/test/Dialect/Vector/
Dinvalid.mlir263 …%0 = vector.transfer_read %arg0[%c3, %c3], %cst { permutation_map = affine_map<()->(0)> } : memref…
292 …%0 = vector.transfer_read %arg0[%c3, %c3, %c3], %cst { permutation_map = affine_map<()->(0)> } : m…
300 …// expected-error@+1 {{requires a permutation_map with input dims of the same rank as the memref t…
301 …%0 = vector.transfer_read %arg0[%c3, %c3], %cst {permutation_map = affine_map<(d0)->(d0)>} : memre…
309 …// expected-error@+1 {{requires a permutation_map with result dims of the same rank as the vector …
310 …%0 = vector.transfer_read %arg0[%c3, %c3], %cst {permutation_map = affine_map<(d0, d1)->(d0, d1)>}…
318 …// expected-error@+1 {{requires a projected permutation_map (at most one dim or the zero constant …
319 …%0 = vector.transfer_read %arg0[%c3, %c3], %cst {permutation_map = affine_map<(d0, d1)->(d0 + d1)>…
327 …// expected-error@+1 {{requires a projected permutation_map (at most one dim or the zero constant …
328 …%0 = vector.transfer_read %arg0[%c3, %c3], %cst {permutation_map = affine_map<(d0, d1)->(d0 + 1)>}…
[all …]
Dops.mlir17 …%0 = vector.transfer_read %arg0[%c3, %c3], %f0 {permutation_map = affine_map<(d0, d1)->(d0)>} : me…
19 …%1 = vector.transfer_read %arg0[%c3, %c3], %f0 {permutation_map = affine_map<(d0, d1)->(d1, d0)>} …
21 …%2 = vector.transfer_read %arg0[%c3, %c3], %cst {permutation_map = affine_map<(d0, d1)->(d0)>} : m…
23 …%3 = vector.transfer_read %arg0[%c3, %c3], %cst {permutation_map = affine_map<(d0, d1)->(d1)>} : m…
25 …%4 = vector.transfer_read %arg1[%c3, %c3], %vf0 {permutation_map = affine_map<(d0, d1)->(d0, d1)>}…
33 …vector.transfer_write %0, %arg0[%c3, %c3] {permutation_map = affine_map<(d0, d1)->(d0)>} : vector<…
35 …vector.transfer_write %1, %arg0[%c3, %c3] {permutation_map = affine_map<(d0, d1)->(d1, d0)>} : vec…
37 …vector.transfer_write %4, %arg1[%c3, %c3] {permutation_map = affine_map<(d0, d1)->(d0, d1)>} : vec…
Dvector-transforms.mlir263 { permutation_map = affine_map<(d0, d1) -> (d0, d1)> }
267 { permutation_map = affine_map<(d0, d1) -> (d0, d1)> }
271 { permutation_map = affine_map<(d0, d1) -> (d0, d1)> }
278 {permutation_map = affine_map<(d0, d1) -> (d0, d1)>}
298 …%4 = vector.transfer_read %0[%arg2, %arg3], %cst {permutation_map = affine_map<(d0, d1) -> (d0, d1…
299 …%5 = vector.transfer_read %1[%arg2, %arg3], %cst {permutation_map = affine_map<(d0, d1) -> (d0, d1…
301 …vector.transfer_write %6, %2[%arg2, %arg3] {permutation_map = affine_map<(d0, d1) -> (d0, d1)>} : …
441 {permutation_map = affine_map<(d0, d1, d2) -> (d1, d2)>}
453 {permutation_map = affine_map<(d0, d1, d2) -> (d1, d2)>}
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_ordering_amd_op.cc119 IndicesMap permutation_map( in Compute() local
122 permutation_map = permutation_matrix.indices(); in Compute()
/external/llvm-project/mlir/lib/Conversion/VectorToSCF/
DVectorToSCF.cpp226 applyMapToValues(rewriter, xferOp.getLoc(), xferOp.permutation_map(), in emitInBoundsCondition()
430 auto exprs = transfer.permutation_map().getResults(); in computeCoalescedIndex()
471 applyMapToValues(rewriter, transfer.getLoc(), transfer.permutation_map(), in emitWithBoundsChecks()
487 auto exprs = transfer.permutation_map().getResults(); in emitWithBoundsChecks()
547 transfer.permutation_map().isMinorIdentity()) { in matchAndRewrite()
632 transfer.permutation_map().isMinorIdentity()) { in matchAndRewrite()
/external/llvm-project/mlir/test/Conversion/VectorToSCF/
Dvector-to-loops.mlir10 …%f1 = vector.transfer_read %A[%i0, %i1], %f0 {permutation_map = affine_map<(d0, d1) -> (d0)>} : me…
12 …%f2 = vector.transfer_read %A[%i0, %ip1], %f0 {permutation_map = affine_map<(d0, d1) -> (d0)>} : m…
14 …%f3 = vector.transfer_read %A[%i0, %ip2], %f0 {permutation_map = affine_map<(d0, d1) -> (d0)>} : m…
16 …%f4 = vector.transfer_read %A[%i0, %ip3], %f0 {permutation_map = affine_map<(d0, d1) -> (d0)>} : m…
41 …%f1 = vector.transfer_read %A[%i0, %i1, %i2, %i3, %i4], %f0 {permutation_map = affine_map<(d0, d1,…
43 …%f2 = vector.transfer_read %A[%i0, %i1, %i2, %i3p1, %i4], %f0 {permutation_map = affine_map<(d0, d…
106 …%f = vector.transfer_read %A[%i0, %i1, %i2, %i3], %f0 {permutation_map = affine_map<(d0, d1, d2, d…
166 …vector.transfer_write %f1, %A[%i0, %i1, %i2, %i3] {permutation_map = affine_map<(d0, d1, d2, d3) -…
364 { permutation_map = affine_map<(d0, d1, d2, d3) -> (d2, d3)> }
394 { permutation_map = affine_map<(d0, d1, d2, d3) -> (d2, d3)> }
/external/llvm-project/mlir/lib/Conversion/StandardToSPIRV/
DLegalizeStandardForSPIRV.cpp71 loadOp.permutation_map(), loadOp.padding(), loadOp.maskedAttr()); in replaceOp()
88 sourceIndices, tranferWriteOp.permutation_map(), in replaceOp()
/external/llvm-project/mlir/integration_test/Dialect/Standard/CPU/
Dtest-ceil-floor-pos-neg.mlir9 {permutation_map = affine_map<(d0) -> (d0)>} :
/external/llvm-project/mlir/lib/Conversion/VectorToROCDL/
DVectorToROCDL.cpp75 if (!xferOp.permutation_map().isMinorIdentity()) in matchAndRewrite()
/external/llvm-project/mlir/test/Conversion/VectorToLLVM/
Dvector-mask-to-llvm.mlir46 …%f = vector.transfer_read %A[%i], %d {permutation_map = affine_map<(d0) -> (d0)>} : memref<?xf32>,…

12