Searched defs:SetGlobalTPUArrayOpKernel (Results 1 – 1 of 1) sorted by relevance
275 class SetGlobalTPUArrayOpKernel : public OpKernel { class277 explicit SetGlobalTPUArrayOpKernel(OpKernelConstruction* ctx) in SetGlobalTPUArrayOpKernel() function in tensorflow::dtensor::SetGlobalTPUArrayOpKernel