Searched refs:from_ops (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 53 def _MarkReachedOps(from_ops, reached_ops, func_graphs): argument
63 queue.extend(from_ops)
73 def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops, func_graphs, argument
99 _MarkReachedOps(from_ops, reached_ops, func_graphs)
259 def _StopOps(from_ops, stop_gradient_ops, pending_count, xs_set): argument
281 for op in from_ops:
359 def _RaiseNoGradWrtInitialLoopValError(op, from_ops, xs_set): argument
370 if curr_op in from_ops:
549 from_ops = [t.op for t in xs]
552 to_ops, from_ops, colocate_gradients_with_ops, func_graphs, xs_set)
[all …]
|