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