Searched defs:local_step (Results 1 – 2 of 2) sorted by relevance
51 void TryApplyGrad(int64 local_step, OpKernelContext* ctx) override { in TryApplyGrad()
1268 def apply_grad(self, grad, local_step=0, name=None): argument1349 def apply_indexed_slices_grad(self, grad, local_step=0, name=None): argument1377 local_step=0, argument