Searched refs:XlaLocalLaunchOp (Results 1 – 5 of 5) sorted by relevance
133 class XlaLocalLaunchOp : public XlaLocalLaunchBase {135 explicit XlaLocalLaunchOp(OpKernelConstruction* ctx);136 ~XlaLocalLaunchOp() override;139 TF_DISALLOW_COPY_AND_ASSIGN(XlaLocalLaunchOp);
425 XlaLocalLaunchOp::XlaLocalLaunchOp(OpKernelConstruction* ctx) in XlaLocalLaunchOp() function in tensorflow::XlaLocalLaunchOp429 XlaLocalLaunchOp::~XlaLocalLaunchOp() { in ~XlaLocalLaunchOp()556 REGISTER_KERNEL_BUILDER(Name("XlaLaunch").Device(DEVICE_CPU), XlaLocalLaunchOp);562 XlaLocalLaunchOp);
80 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_INTERPRETER, XlaLocalLaunchOp,
91 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_CPU, XlaLocalLaunchOp, kAllXlaCpuTypes);
124 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_GPU, XlaLocalLaunchOp, kAllXlaGpuTypes);