Searched defs:OpAdaptor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_tf_framework.cc | 37 func::FuncOp func, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
|
| D | tf_framework_legalize_to_llvm.cc | 461 NullContextOp op, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
473 NullMemRefOp null_memref_op, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 199 lmhlo::ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
370 lmhlo::ReduceWindowOp reduceWindowOp, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
495 lmhlo::SelectAndScatterOp sAndSOp, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
|
| D | lhlo_legalize_to_gpu.cc | 55 ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1176 mhlo::ConstantOp constOp, OpAdaptor /*adaptor*/, in matchAndRewrite() argument
|