Searched refs:SetGlobalTPUArrayOp (Results 1 – 2 of 2) sorted by relevance
131 class SetGlobalTPUArrayOp : public OpKernel {133 explicit SetGlobalTPUArrayOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in SetGlobalTPUArrayOp() function137 ~SetGlobalTPUArrayOp() override {} in ~SetGlobalTPUArrayOp()141 SetGlobalTPUArrayOp(const SetGlobalTPUArrayOp&) = delete;142 SetGlobalTPUArrayOp& operator=(const SetGlobalTPUArrayOp&) = delete;
370 void SetGlobalTPUArrayOp::Compute(OpKernelContext* ctx) { in Compute()426 SetGlobalTPUArrayOp);