Searched defs:outer_grad_state (Results 1 – 2 of 2) sorted by relevance
122 def __init__(self, forward_ctxt, outer_grad_state): argument209 def outer_grad_state(self): member in _GradLoopState
1841 def AddForwardLoopCounter(self, outer_grad_state): argument1890 def AddBackpropLoopCounter(self, count, outer_grad_state): argument