Searched refs:grad_context (Results 1 – 3 of 3) sorted by relevance
180 outer_grad_ctxt = outer_grad_state.grad_context238 def grad_context(self): member in _GradLoopState414 self.grad_context.Enter()430 self.grad_context.Exit()431 parallel_iterations = self.grad_context.parallel_iterations557 grad_state.grad_context.Enter()563 grad_state.grad_context.Exit()620 outer_grad_state.grad_context.Enter()622 outer_grad_state.grad_context.Exit()631 outer_grad_ctxt = outer_grad_state.grad_context[all …]
119 grad_ctxt = grad_state.grad_context
131 grad_context = control_flow_util.GetOutputContext(grad.op)133 if dim_context != grad_context: