D | xla_device_ops.h | 39 // Dummy OpKernel, used for kernels assigned to an XLA device that should be 42 // operator on an XLA device but the compiler did not compile it. 58 #define REGISTER_XLA_LAUNCH_KERNEL(DEVICE, KERNEL, TYPES) \ argument 60 .Device(DEVICE) \ 65 #define REGISTER_XLA_COMPILE_KERNEL(DEVICE, KERNEL, TYPES) \ argument 67 .Device(DEVICE) \ 74 #define REGISTER_XLA_RUN_KERNEL(DEVICE, KERNEL, TYPES) \ argument 75 REGISTER_KERNEL_BUILDER(Name("_XlaRun").Device(DEVICE), KERNEL); 77 #define REGISTER_XLA_DEVICE_KERNELS(DEVICE, TYPES) \ argument 79 Name("Const").Device(DEVICE).TypeConstraint("dtype", TYPES), \ [all …]
|