Searched refs:XlaRunOp (Results 1 – 5 of 5) sorted by relevance
120 class XlaRunOp : public OpKernel {122 explicit XlaRunOp(OpKernelConstruction* ctx);
455 XlaRunOp::XlaRunOp(OpKernelConstruction* ctx) in XlaRunOp() function in tensorflow::XlaRunOp458 void XlaRunOp::Compute(OpKernelContext* ctx) { in Compute()574 REGISTER_KERNEL_BUILDER(Name("_XlaRun").Device(DEVICE_CPU), XlaRunOp);576 XlaRunOp);
117 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_CPU, XlaRunOp, kAllXlaCpuTypes);
463 REGISTER_XLA_RUN_KERNEL(DEVICE_TPU_NODE, XlaRunOp, kTpuAllTypes); in RegisterTpuNodeDevice()479 REGISTER_XLA_RUN_KERNEL(DEVICE_TPU_NODE, XlaRunOp, kTpuAllTypes);
182 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_GPU, XlaRunOp, kAllXlaGpuTypes);