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