Searched refs:func_graphs (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
70 queue.extend(_Consumers(output, func_graphs))
73 def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops, func_graphs, argument
99 _MarkReachedOps(from_ops, reached_ops, func_graphs)
461 def _Consumers(t, func_graphs): argument
473 for func in func_graphs:
476 consumers.extend(_Consumers(placeholder, func_graphs))
504 func_graphs = []
507 func_graphs.append(curr_graph)
552 to_ops, from_ops, colocate_gradients_with_ops, func_graphs, xs_set)
|