Searched refs:XlaCompileOp (Results 1 – 5 of 5) sorted by relevance
142 class XlaCompileOp : public OpKernel {144 explicit XlaCompileOp(OpKernelConstruction* ctx);
433 XlaCompileOp::XlaCompileOp(OpKernelConstruction* ctx) in XlaCompileOp() function in tensorflow::XlaCompileOp441 void XlaCompileOp::Compute(OpKernelContext* ctx) { in Compute()564 REGISTER_KERNEL_BUILDER(Name("_XlaCompile").Device(DEVICE_CPU), XlaCompileOp);571 XlaCompileOp);
82 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_INTERPRETER, XlaCompileOp,
92 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_CPU, XlaCompileOp, kAllXlaCpuTypes);
125 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_GPU, XlaCompileOp, kAllXlaGpuTypes);