Searched refs:XlaRunOp (Results 1 – 5 of 5) sorted by relevance
169 class XlaRunOp : public OpKernel {171 explicit XlaRunOp(OpKernelConstruction* ctx);
509 XlaRunOp::XlaRunOp(OpKernelConstruction* ctx) in XlaRunOp() function in tensorflow::XlaRunOp512 void XlaRunOp::Compute(OpKernelContext* ctx) { in Compute()573 REGISTER_KERNEL_BUILDER(Name("_XlaRun").Device(DEVICE_CPU), XlaRunOp);574 REGISTER_KERNEL_BUILDER(Name("_XlaRun").Device(DEVICE_GPU), XlaRunOp);
84 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_INTERPRETER, XlaRunOp, kExecAllTypes);
93 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_CPU, XlaRunOp, kAllXlaCpuTypes);
126 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_GPU, XlaRunOp, kAllXlaGpuTypes);