HomeSort by relevance Sort by last modified time
    Searched refs:reached_ops (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/ops/
gradients_impl.py 108 def _MarkReachedOps(from_ops, reached_ops):
113 reached_ops: list of booleans, indexed by operation id.
119 if not reached_ops[op._id]:
120 reached_ops[op._id] = True
125 def _GatherInputs(to_ops, reached_ops):
126 """List all inputs of to_ops that are in reached_ops.
130 reached_ops: list of booleans, indexed by operation id.
133 The list of all inputs of to_ops that are in reached_ops.
142 if reached_ops[op._id]:
145 reached_ops[op._id] = Fals
    [all...]
gradients_test.py 71 reached_ops = [False] * (graph._last_id + 1)
75 reached_ops[op._id] = True
76 gradients_impl._MarkReachedOps(from_ops, reached_ops)
77 between_ops = gradients_impl._GatherInputs(to_ops, reached_ops)

Completed in 83 milliseconds