Searched refs:AddBackpropLoopCounter (Results 1 – 2 of 2) sorted by relevance
186 self._grad_index = self._grad_context.AddBackpropLoopCounter(199 self._grad_index = self._grad_context.AddBackpropLoopCounter(
1848 def AddBackpropLoopCounter(self, count, outer_grad_state): member in WhileContext