Searched refs:XlaLocalLaunchOp (Results 1 – 5 of 5) sorted by relevance
80 class XlaLocalLaunchOp : public XlaLocalLaunchBase {82 explicit XlaLocalLaunchOp(OpKernelConstruction* ctx);83 ~XlaLocalLaunchOp() override;86 TF_DISALLOW_COPY_AND_ASSIGN(XlaLocalLaunchOp);
353 XlaLocalLaunchOp::XlaLocalLaunchOp(OpKernelConstruction* ctx) in XlaLocalLaunchOp() function in tensorflow::XlaLocalLaunchOp357 XlaLocalLaunchOp::~XlaLocalLaunchOp() { in ~XlaLocalLaunchOp()557 REGISTER_KERNEL_BUILDER(Name("XlaLaunch").Device(DEVICE_CPU), XlaLocalLaunchOp);563 XlaLocalLaunchOp);
115 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_CPU, XlaLocalLaunchOp, kAllXlaCpuTypes);
461 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_TPU_NODE, XlaLocalLaunchOp, kTpuAllTypes); in RegisterTpuNodeDevice()477 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_TPU_NODE, XlaLocalLaunchOp, kTpuAllTypes);
180 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_GPU, XlaLocalLaunchOp, kAllXlaGpuTypes);