Searched refs:extra_dependencies (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | topological_sort.cc | 36 const absl::Span<const TopologicalDependency> extra_dependencies) { in MakeEphemeralEdges() argument
38 ephemeral_edges.reserve(extra_dependencies.size()); in MakeEphemeralEdges()
39 for (const auto& dep : extra_dependencies) { in MakeEphemeralEdges()
51 const absl::Span<const TopologicalDependency> extra_dependencies, in ComputeTopologicalOrder() argument
55 graph, MakeEphemeralEdges(extra_dependencies))); in ComputeTopologicalOrder()
115 const absl::Span<const TopologicalDependency> extra_dependencies, in ComputeTopologicalOrder() argument
119 ComputeTopologicalOrder(graph, extra_dependencies, &ready_nodes)); in ComputeTopologicalOrder()
|
| D | topological_sort_test.cc | 175 std::vector<TopologicalDependency> extra_dependencies; in TEST_F() local
176 extra_dependencies.push_back({&graph.node(5), &graph.node(4)}); in TEST_F()
179 TF_EXPECT_OK(ComputeTopologicalOrder(graph, extra_dependencies, &topo_order)); in TEST_F()
195 extra_dependencies.push_back({&graph.node(1), &graph.node(5)}); in TEST_F()
197 ComputeTopologicalOrder(graph, extra_dependencies, &topo_order).ok()); in TEST_F()
|
| D | topological_sort.h | 44 absl::Span<const TopologicalDependency> extra_dependencies,
|
| D | graph_view.h | 416 absl::Span<const TopologicalDependency> extra_dependencies);
|
| D | graph_view.cc | 1428 absl::Span<const TopologicalDependency> extra_dependencies) { in SortTopologically() argument
1440 for (const auto& extra_dependency : extra_dependencies) { in SortTopologically()
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | source_writer.cc | 168 const std::list<Type>* extra_dependencies, in BeginType() argument
175 if (extra_dependencies != nullptr) { in BeginType()
176 for (const Type& t : *extra_dependencies) { in BeginType()
|
| D | source_writer.h | 129 const std::list<Type>* extra_dependencies = nullptr,
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save.py | 117 extra_dependencies = self.list_extra_dependencies(obj)
118 extra_dependencies.update(self._extra_dependencies.get(obj, {}))
123 if name in extra_dependencies:
135 extra_dependencies.keys()))
136 yield base.TrackableReference(name, extra_dependencies[name])
139 for name, dep in extra_dependencies.items():
|