|
|
@ -194,6 +194,8 @@ class OpKernelRegistrar : public Registrar {
|
|
|
|
USE_OP_DEVICE_KERNEL(op_type, GPU)
|
|
|
|
USE_OP_DEVICE_KERNEL(op_type, GPU)
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define USE_NO_KERNEL_OP(op_type) USE_OP_ITSELF(op_type);
|
|
|
|
|
|
|
|
|
|
|
|
#define USE_CPU_ONLY_OP(op_type) \
|
|
|
|
#define USE_CPU_ONLY_OP(op_type) \
|
|
|
|
USE_OP_ITSELF(op_type); \
|
|
|
|
USE_OP_ITSELF(op_type); \
|
|
|
|
USE_OP_DEVICE_KERNEL(op_type, CPU);
|
|
|
|
USE_OP_DEVICE_KERNEL(op_type, CPU);
|
|
|
|