Searched refs:ShutdownDistributedTpuOp (Results 1 – 2 of 2) sorted by relevance
86 class ShutdownDistributedTpuOp : public OpKernel {88 explicit ShutdownDistributedTpuOp(OpKernelConstruction* ctx) in ShutdownDistributedTpuOp() function93 ~ShutdownDistributedTpuOp() override {} in ~ShutdownDistributedTpuOp()97 ShutdownDistributedTpuOp(const ShutdownDistributedTpuOp&) = delete;98 ShutdownDistributedTpuOp& operator=(const ShutdownDistributedTpuOp&) = delete;
235 void ShutdownDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute()418 ShutdownDistributedTpuOp);