D | base_op.h | 24 #include "tensorflow/core/platform/errors.h" 140 #define MLIR_FUNCTION(tf_op, platform, input_type, output_type) \ argument 141 _mlir_ciface_##tf_op##_##platform##_##input_type##_##output_type 143 #define MLIR_OP(tf_op, platform, input_type, output_type) \ argument 144 Mlir##tf_op##platform##input_type##output_type##Op 148 #define REGISTER_ALIASED_KERNEL(tf_op, mlir_op, platform, input_type, \ argument 152 .Device(DEVICE_##platform) \ 155 MLIR_OP(mlir_op, platform, input_type, output_type)); 157 #define REGISTER_KERNEL(tf_op, platform, input_type, output_type, \ argument 159 REGISTER_ALIASED_KERNEL(tf_op, tf_op, platform, input_type, output_type, \ [all …]
|