Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:grad_state (Results 1 - 6 of 6) sorted by null

/external/tensorflow/tensorflow/python/ops/
control_flow_grad.py	48 merge_grad = grad_ctxt.grad_state.switch_map.get(op) 66 grad_ctxt.grad_state.switch_map[op] = merge_grad 110 if grad_ctxt and grad_ctxt.grad_state: 115 grad_state = grad_ctxt.grad_state 116 real_pred = grad_state.history_map.get(pred.name) 119 grad_ctxt = grad_state.grad_context 121 history_pred = grad_state.AddForwardAccumulator(pred) 126 real_pred = grad_state.AddBackpropAccumulatedValue(history_pred, pred) 127 grad_state.history_map[pred.name] = real_pre     [all...]
control_flow_util.py	298 elif (while_ctxt.grad_state and 299 IsContainingContext(while_ctxt.grad_state.forward_context, 308 elif (while_ctxt.grad_state and 309 while_ctxt.grad_state.forward_context is 315 elif (input_while_ctxt.grad_state and 316 input_while_ctxt.grad_state.forward_context is while_ctxt): 321 elif (input_while_ctxt.grad_state and 322 input_ctxt.grad_state.forward_context.grad_state and 323 input_ctxt.grad_state.forward_context.grad_state.forward_context i     [all...]
control_flow_ops.py	727 grad_state=self) 741 grad_state=self) 1425 def grad_state(self): member in class:ControlFlowContext 1623 def grad_state(self): member in class:CondContext 2322 def grad_state(self): member in class:WhileContext     [all...]
gradients_util.py	804 grad_state = loop_state.GetGradState(x.op, before=False) 805 grad_state.deferred_exits.append(x) 806 grad_state.pending_exits_count -= 1 807 if grad_state.pending_exits_count == 0: 810 for y in grad_state.deferred_exits: 815 grad_state.unused_exits.append(y) 819 for y in grad_state.unused_exits: 825 for y in grad_state.unused_exits:     [all...]
/external/tensorflow/tensorflow/contrib/compiler/
xla.py	267 def grad_state(self): member in class:XLACompileContext 270 # grad_state outside the XLACompileContext affect the graph inside so the 271 # grad_state should be as if this is the top-level gradient state.
/external/tensorflow/tensorflow/python/tpu/
tpu.py	454 def grad_state(self): member in class:TPUReplicateContext 457 # grad_state outside the TPUReplicateContext affect the graph inside so the 458 # grad_state should be as if this is the top-level gradient state. 1482 def grad_state(self): member in class:_TPUInferenceContext     [all...]

Completed in 1480 milliseconds