Searched defs:outer_grad_state (Results 1 – 1 of 1) sorted by relevance
661 def __init__(self, forward_ctxt, outer_grad_state): argument748 def outer_grad_state(self): member in GradLoopState2541 def AddForwardLoopCounter(self, outer_grad_state): argument2590 def AddBackpropLoopCounter(self, count, outer_grad_state): argument