Searched refs:GradLoopState (Results 1 – 1 of 1) sorted by relevance
647 class GradLoopState(object): class1129 grad_state = GradLoopState(forward_ctxt, outer_grad_state)