Searched refs:InitializeHostForDistributedTpuOp (Results 1 – 2 of 2) sorted by relevance
105 class InitializeHostForDistributedTpuOp : public OpKernel {107 explicit InitializeHostForDistributedTpuOp(OpKernelConstruction* ctx) in InitializeHostForDistributedTpuOp() function116 ~InitializeHostForDistributedTpuOp() override {} in ~InitializeHostForDistributedTpuOp()120 InitializeHostForDistributedTpuOp(const InitializeHostForDistributedTpuOp&) =122 InitializeHostForDistributedTpuOp& operator=(123 const InitializeHostForDistributedTpuOp&) = delete;
251 void InitializeHostForDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute()423 InitializeHostForDistributedTpuOp);