Searched refs:XlaCompileOp (Results 1 – 5 of 5) sorted by relevance
89 class XlaCompileOp : public OpKernel {91 explicit XlaCompileOp(OpKernelConstruction* ctx);
361 XlaCompileOp::XlaCompileOp(OpKernelConstruction* ctx) in XlaCompileOp() function in tensorflow::XlaCompileOp370 void XlaCompileOp::Compute(OpKernelContext* ctx) { in Compute()565 REGISTER_KERNEL_BUILDER(Name("_XlaCompile").Device(DEVICE_CPU), XlaCompileOp);572 XlaCompileOp);
116 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_CPU, XlaCompileOp, kAllXlaCpuTypes);
462 REGISTER_XLA_COMPILE_KERNEL(DEVICE_TPU_NODE, XlaCompileOp, kTpuAllTypes); in RegisterTpuNodeDevice()478 REGISTER_XLA_COMPILE_KERNEL(DEVICE_TPU_NODE, XlaCompileOp, kTpuAllTypes);
181 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_GPU, XlaCompileOp, kAllXlaGpuTypes);