HomeSort by relevance Sort by last modified time
    Searched refs:grad_state (Results 1 - 5 of 5) 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
105 if grad_ctxt and grad_ctxt.grad_state:
110 grad_state = grad_ctxt.grad_state
111 real_pred = grad_state.history_map.get(pred.name)
114 grad_ctxt = grad_state.grad_context
116 history_pred = grad_state.AddForwardAccumulator(pred)
121 real_pred = grad_state.AddBackpropAccumulatedValue(history_pred, pred)
122 grad_state.history_map[pred.name] = real_pre
    [all...]
control_flow_util.py 211 elif (while_ctxt.grad_state and
212 IsContainingContext(while_ctxt.grad_state.forward_context,
221 elif (while_ctxt.grad_state and
222 while_ctxt.grad_state.forward_context is
228 elif (input_while_ctxt.grad_state and
229 input_while_ctxt.grad_state.forward_context is while_ctxt):
234 elif (input_while_ctxt.grad_state and
235 input_ctxt.grad_state.forward_context.grad_state and
236 input_ctxt.grad_state.forward_context.grad_state.forward_context i
    [all...]
control_flow_ops.py 1551 def grad_state(self): member in class:ControlFlowContext
1735 def grad_state(self): member in class:CondContext
2350 def grad_state(self): member in class:WhileContext
    [all...]
gradients_impl.py 679 grad_state = loop_state.GetGradState(x.op, before=False)
680 grad_state.deferred_exits.append(x)
681 grad_state.pending_exits_count -= 1
682 if grad_state.pending_exits_count == 0:
685 for y in grad_state.deferred_exits:
690 grad_state.unused_exits.append(y)
694 for y in grad_state.unused_exits:
700 for y in grad_state.unused_exits:
    [all...]
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu.py 176 def grad_state(self): member in class:TPUReplicateContext
179 # grad_state outside the TPUReplicateContext affect the graph inside so the
180 # grad_state should be as if this is the top-level gradient state.

Completed in 1672 milliseconds