Searched refs:grad_context (Results 1 – 3 of 3) sorted by relevance
719 outer_grad_ctxt = outer_grad_state.grad_context777 def grad_context(self): member in GradLoopState953 self.grad_context.Enter()968 self.grad_context.Exit()969 parallel_iterations = self.grad_context.parallel_iterations1103 grad_state.grad_context.Enter()1109 grad_state.grad_context.Exit()1166 outer_grad_state.grad_context.Enter()1168 outer_grad_state.grad_context.Exit()1177 outer_grad_ctxt = outer_grad_state.grad_context[all …]
119 grad_ctxt = grad_state.grad_context
125 grad_context = control_flow_util.GetOutputContext(grad.op)127 if dim_context != grad_context: