Searched defs:mlir_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_cpu_op.h | 32 #define REGISTER_ALIASED_CPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
|
| D | base_gpu_op.h | 32 #define REGISTER_ALIASED_GPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
|
| D | base_op.h | 152 #define REGISTER_ALIASED_KERNEL(tf_op, mlir_op, platform, input_type, \ argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 258 #define MakeAndReturn(mlir_op) \ in ImportInstructionImpl() argument
737 #define NoAttributeCase(hlo_op_code, mlir_op) \ in ImportInstructionImpl() argument
|