Searched refs:TPUReshardVariablesOpKernel (Results 1 – 2 of 2) sorted by relevance
51 TPUReshardVariablesOpKernel::TPUReshardVariablesOpKernel( in TPUReshardVariablesOpKernel() function in tensorflow::TPUReshardVariablesOpKernel57 void TPUReshardVariablesOpKernel::ComputeAsync(OpKernelContext* context, in ComputeAsync()65 Status TPUReshardVariablesOpKernel::DoWork(OpKernelContext* context) { in DoWork()121 Status TPUReshardVariablesOpKernel::DoTpuExecute( in DoTpuExecute()273 TPUReshardVariablesOpKernel::~TPUReshardVariablesOpKernel() = default;280 TPUReshardVariablesOpKernel);
33 class TPUReshardVariablesOpKernel : public AsyncOpKernel {35 explicit TPUReshardVariablesOpKernel(OpKernelConstruction* context);36 ~TPUReshardVariablesOpKernel() override;