Searched refs:TryApplyGrad (Results 1 – 3 of 3) sorted by relevance
52 void TryApplyGrad(int64 local_step, OpKernelContext* ctx) override { in TryApplyGrad() function
59 virtual void TryApplyGrad(int64 local_step, OpKernelContext* ctx) = 0;
195 accumulator->TryApplyGrad(local_step_tensor->scalar<int64>()(), ctx); in Compute()