D | xla_device_ops.h | 56 #define REGISTER_XLA_LAUNCH_KERNEL(DEVICE, KERNEL, TYPES) \ argument 61 KERNEL); 63 #define REGISTER_XLA_COMPILE_KERNEL(DEVICE, KERNEL, TYPES) \ argument 70 KERNEL); 72 #define REGISTER_XLA_RUN_KERNEL(DEVICE, KERNEL, TYPES) \ argument 73 REGISTER_KERNEL_BUILDER(Name("_XlaRun").Device(DEVICE), KERNEL);
|