Home
last modified time | relevance | path

Searched refs:grad_context (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_state.py180 outer_grad_ctxt = outer_grad_state.grad_context
238 def grad_context(self): member in _GradLoopState
414 self.grad_context.Enter()
430 self.grad_context.Exit()
431 parallel_iterations = self.grad_context.parallel_iterations
557 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 …]
Dcontrol_flow_grad.py119 grad_ctxt = grad_state.grad_context
Darray_grad.py131 grad_context = control_flow_util.GetOutputContext(grad.op)
133 if dim_context != grad_context: