Searched refs:XlaLocalLaunchBase (Results 1 – 3 of 3) sorted by relevance
47 class XlaLocalLaunchBase : public OpKernel {49 XlaLocalLaunchBase(OpKernelConstruction* ctx,53 XlaLocalLaunchBase(const XlaLocalLaunchBase&) = delete;54 XlaLocalLaunchBase& operator=(const XlaLocalLaunchBase&) = delete;55 ~XlaLocalLaunchBase() override = default;80 class XlaLocalLaunchOp : public XlaLocalLaunchBase {
152 XlaLocalLaunchBase::XlaLocalLaunchBase(OpKernelConstruction* ctx, in XlaLocalLaunchBase() function in tensorflow::XlaLocalLaunchBase219 void XlaLocalLaunchBase::Compute(OpKernelContext* ctx) { in Compute()354 : XlaLocalLaunchBase(ctx, ConstantsVector(ctx), ResourcesVector(ctx), in XlaLocalLaunchOp()
166 *kernel = absl::make_unique<XlaLocalLaunchBase>( in CreateXlaKernel()