Searched refs:func_graphs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 103 def _MarkReachedOps(from_ops, reached_ops, func_graphs): argument
120 queue.extend(_Consumers(output, func_graphs))
123 def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops, func_graphs, argument
149 _MarkReachedOps(from_ops, reached_ops, func_graphs)
527 def _Consumers(t, func_graphs): argument
539 for func in func_graphs:
542 consumers.extend(_Consumers(placeholder, func_graphs))
570 func_graphs = []
573 func_graphs.append(curr_graph)
617 to_ops, from_ops, colocate_gradients_with_ops, func_graphs, xs)
|