| 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 …]
 
 |