Home
last modified time | relevance | path

Searched refs:batch_matmul (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Droundtrip.mlir310 linalg.batch_matmul ins(%a3, %b3: memref<?x?x?xf32>, memref<?x?x?xf32>)
312 linalg.batch_matmul ins(%ta3, %tb3: tensor<?x?x?xf32>, tensor<?x?x?xf32>)
314 %res1 = linalg.batch_matmul
318 %res2 = linalg.batch_matmul
325 // CHECK: linalg.batch_matmul
326 // CHECK: linalg.batch_matmul
327 // CHECK: linalg.batch_matmul
328 // CHECK: linalg.batch_matmul
H A Dreshape_control_fusion.mlir52 %1 = linalg.batch_matmul ins(%arg0, %arg1 : tensor<1x?x?xf32>, tensor<1x?x?xf32>)
61 // CHECK: linalg.batch_matmul
H A Daffine.mlir21 linalg.batch_matmul ins(%A, %B: memref<?x?x?xf32>, memref<?x?x?xf32>)
H A Dinvalid.mlir315 linalg.batch_matmul ins(%a3, %b3: memref<?x?x?xf32>, memref<?x?xf32>)
H A Dloops.mlir651 linalg.batch_matmul ins(%A, %B : memref<?x?x?xf32>, memref<?x?x?xf32>)
H A Dvectorization.mlir44 linalg.batch_matmul
/llvm-project-15.0.7/mlir/test/Conversion/TosaToLinalg/
H A Dtosa-to-linalg-named.mlir8 …// CHECK: linalg.batch_matmul ins(%arg0, %arg1 : tensor<1x5x3xf32>, tensor<1x3x6xf32>) outs([[FILL…
37 …// CHECK: linalg.batch_matmul ins(%arg0, %arg1 : tensor<?x5x3xf32>, tensor<?x3x6xf32>) outs(%[[FIL…
51 …// CHECK: linalg.batch_matmul ins(%arg0, %arg1 : tensor<1x5x3xf32>, tensor<1x3x?xf32>) outs(%[[FIL…
63 …// CHECK: linalg.batch_matmul ins(%arg0, %arg1 : tensor<1x5x?xf32>, tensor<1x?x6xf32>) outs(%[[FIL…
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py120 def batch_matmul(A=TensorDef(T1, Batch, S.M, S.K), function
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml483 name: batch_matmul