| /llvm-project-15.0.7/mlir/test/Dialect/SparseTensor/ |
| H A D | sparse_parallel.mlir | 131 // CHECK-PAR0-LABEL: func @matvec
136 // CHECK-PAR1-LABEL: func @matvec
141 // CHECK-PAR2-LABEL: func @matvec
146 // CHECK-PAR3-LABEL: func @matvec
151 // CHECK-PAR4-LABEL: func @matvec
156 func.func @matvec(%arga: tensor<16x32xf32, #CSR>,
|
| H A D | sparse_lower_inplace.mlir | 23 // CHECK-HIR-LABEL: func @matvec(
54 // CHECK-MIR-LABEL: func @matvec(
85 // CHECK-LIR-LABEL: func @matvec(
113 func.func @matvec(%arga: tensor<32x64xf64, #CSR>,
|
| H A D | sparse_lower.mlir | 23 // CHECK-HIR-LABEL: func @matvec(
54 // CHECK-MIR-LABEL: func @matvec(
85 // CHECK-LIR-LABEL: func @matvec(
113 func.func @matvec(%arga: tensor<32x64xf64, #CSR>,
|
| H A D | sparse_lower_col.mlir | 26 // CHECK-HIR-LABEL: func @matvec(
56 // CHECK-MIR-LABEL: func @matvec(
86 // CHECK-LIR-LABEL: func @matvec(
113 func.func @matvec(%arga: tensor<32x64xf64, #CSC>,
|
| H A D | sparse_2d.mlir | 860 // CHECK-LABEL: func @matvec(
890 func.func @matvec(%argA: tensor<16x32xf32, #Tds>, %argb: tensor<32xf32>, %argx: tensor<16xf32>) -> …
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/Linalg/CPU/ |
| H A D | matmul-vs-matvec.mlir | 21 func.func @matvec(%A: memref<?x?xf32>, %B: memref<?x?xf32>) -> (memref<?x?xf32>) {
33 linalg.matvec ins(%A, %b: memref<?x?xf32>, memref<?xf32, offset: ?, strides: [?]>)
53 %C2 = call @matvec(%A, %B) : (memref<?x?xf32>, memref<?x?xf32>) -> memref<?x?xf32>
59 cf.assert %c, "Matmul does not produce same output as matvec"
|
| /llvm-project-15.0.7/mlir/test/Dialect/Linalg/ |
| H A D | transform-patterns.mlir | 33 func.func @matvec(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
36 linalg.matvec
42 // CHECK-LABEL: func @matvec
48 // CHECK: linalg.matvec
129 linalg.matvec {__internal_linalg_transform__ = "__with_perm__"}
141 // CHECK: linalg.matvec
|
| H A D | tile.mlir | 158 func.func @matvec(%arg0: memref<?x?xf32, offset: ?, strides: [?, 1]>, %arg1: memref<?xf32, offset: …
159 linalg.matvec
165 // TILE-2-LABEL: func @matvec(
178 // TILE-2: linalg.matvec ins(%[[sAi]], %{{.*}} outs(%[[sCi]]
180 // TILE-02-LABEL: func @matvec(
193 // TILE-02: linalg.matvec ins(%[[sAj]], %[[sBj]]{{.*}} outs(%{{.*}}
195 // TILE-002-LABEL: func @matvec(
201 // TILE-234-LABEL: func @matvec(
220 // TILE-234: linalg.matvec ins(%[[sAij]], %[[sBj]]{{.*}} outs(%[[sCi]]
|
| H A D | roundtrip.mlir | 41 linalg.matvec ins(%arg0, %arg1: memref<?x?xf32, offset: ?, strides: [?, 1]>,
54 // CHECK: linalg.matvec
|
| H A D | loops.mlir | 62 func.func @matvec(%arg0: memref<?xi8>, %M: index, %N: index) {
68 linalg.matvec ins(%2, %3: memref<?x?xf32>, memref<?xf32>)
72 // CHECK-LABEL: func @matvec(%{{.*}}: memref<?xi8>,
87 // CHECKPARALLEL-LABEL: func @matvec(%{{.*}}: memref<?xi8>,
|
| H A D | drop-unit-extent-dims.mlir | 803 #matvec = {
816 %1 = linalg.generic #matvec
|
| H A D | vectorization.mlir | 22 linalg.matvec ins(%A, %B: memref<1584x1584xf32>, memref<1584xf32>)
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/CPU/ |
| H A D | sparse_matvec.mlir | 26 #matvec = {
50 %0 = linalg.generic #matvec
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Transforms/ |
| H A D | Passes.td | 28 #matvec = {
42 %0 = linalg.generic #matvec
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransforms.cpp | 1539 FailureOr<Value> matvec() { in matvec() function
1625 FailureOr<Value> matvecRes = e.matvec(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/ |
| H A D | core_named_ops.py | 155 def matvec(A=TensorDef(T1, S.M, S.N), function
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgNamedStructuredOps.yaml | 657 name: matvec
|