Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/test/Integration/Dialect/Linalg/CPU/
H A Dtest-conv-2d-call.mlir21 func.func @conv_2d(%arg0: memref<?x?xf32>, %arg1: memref<?x?xf32>, %arg2: memref<?x?xf32>) {
22 linalg.conv_2d ins (%arg0, %arg1: memref<?x?xf32>, memref<?x?xf32>)
42 …call @conv_2d(%in2D, %filter2D, %out2D) : (memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>) -> ()
/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dtile-conv.mlir9 linalg.conv_2d ins(%arg0, %arg1 : memref<?x?xf32>, memref<?x?xf32>) outs(%arg2 : memref<?x?xf32>)
33 // CHECK: linalg.conv_2d
H A Dfusion.mlir701 …linalg.conv_2d ins(%8, %arg1 : memref<?x?xf32, #map2>, memref<?x?xf32>) outs(%11 : memref<?x?xf32,…
710 // CHECK: linalg.conv_2d
H A Dloops.mlir736 linalg.conv_2d ins(%in, %filter : memref<?x?xf32>, memref<?x?xf32>)
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/CPU/
H A Dsparse_filter_conv2d.mlir21 %0 = linalg.conv_2d
/llvm-project-15.0.7/mlir/test/Dialect/SparseTensor/
H A Dsparse_kernels.mlir197 %0 = linalg.conv_2d
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py225 def conv_2d(I=TensorDef(T1, S.OH + S.KH, S.OW + S.KW), function
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/
H A DAffineOps.td655 func.func @conv_2d(%D : memref<100x100xf32>, %K : memref<3x3xf32>) -> (memref<98x98xf32>) {
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml992 name: conv_2d