Lines Matching refs:grad_context
852 outer_grad_ctxt = outer_grad_state.grad_context
910 def grad_context(self):
1093 self.grad_context.Enter()
1110 self.grad_context.Exit()
1111 parallel_iterations = self.grad_context.parallel_iterations
1247 grad_state.grad_context.Enter()
1253 grad_context.Exit()
1310 outer_grad_state.grad_context.Enter()
1312 outer_grad_state.grad_context.Exit()
1321 outer_grad_ctxt = outer_grad_state.grad_context
1388 grad_state.grad_context.Exit()
1391 grad_state.grad_context.Enter()
1418 grad_state.grad_context.Enter()
1422 grad_state.grad_context.Exit()
1425 outer_grad_ctxt = grad_state.grad_context.outer_context
1435 grad_state.grad_context.Enter()
1437 grad_state.grad_context.Exit()
