OpenGrok
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