Searched refs:DisconnectDistributedTpuChipsOp (Results 1 – 2 of 2) sorted by relevance
149 class DisconnectDistributedTpuChipsOp : public OpKernel {151 explicit DisconnectDistributedTpuChipsOp(OpKernelConstruction* ctx) in DisconnectDistributedTpuChipsOp() function156 ~DisconnectDistributedTpuChipsOp() override {} in ~DisconnectDistributedTpuChipsOp()160 DisconnectDistributedTpuChipsOp(const DisconnectDistributedTpuChipsOp&) =162 DisconnectDistributedTpuChipsOp& operator=(163 const DisconnectDistributedTpuChipsOp&) = delete;
386 void DisconnectDistributedTpuChipsOp::Compute(OpKernelContext* ctx) { in Compute()430 DisconnectDistributedTpuChipsOp);