Searched refs:WaitForDistributedTpuOp (Results 1 – 2 of 2) sorted by relevance
62 class WaitForDistributedTpuOp : public OpKernel {64 explicit WaitForDistributedTpuOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in WaitForDistributedTpuOp() function72 ~WaitForDistributedTpuOp() override {} in ~WaitForDistributedTpuOp()79 WaitForDistributedTpuOp(const WaitForDistributedTpuOp&) = delete;80 WaitForDistributedTpuOp& operator=(const WaitForDistributedTpuOp&) = delete;
145 void WaitForDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute()415 WaitForDistributedTpuOp);