Searched defs:outer_grad_state (Results 1 – 2 of 2) sorted by relevance
118 def __init__(self, forward_ctxt, outer_grad_state): argument205 def outer_grad_state(self): member in _GradLoopState
1799 def AddForwardLoopCounter(self, outer_grad_state): argument1848 def AddBackpropLoopCounter(self, count, outer_grad_state): argument