Searched refs:conv_3d (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/Linalg/CPU/ |
| H A D | test-conv-3d-call.mlir | 21 func.func @conv_3d(%arg0: memref<?x?x?xf32>, %arg1: memref<?x?x?xf32>, %arg2: memref<?x?x?xf32>) {
22 linalg.conv_3d ins (%arg0, %arg1: memref<?x?x?xf32>, memref<?x?x?xf32>)
42 …call @conv_3d(%in3D, %filter3D, %out3D) : (memref<?x?x?xf32>, memref<?x?x?xf32>, memref<?x?x?xf32>…
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/ |
| H A D | core_named_ops.py | 240 def conv_3d(I=TensorDef(T1, S.OD + S.KD, S.OH + S.KH, S.OW + S.KW), function
|
| /llvm-project-15.0.7/mlir/test/Dialect/Linalg/ |
| H A D | loops.mlir | 789 linalg.conv_3d ins(%in, %filter : memref<?x?x?xf32>, memref<?x?x?xf32>)
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgNamedStructuredOps.yaml | 1061 name: conv_3d
|