Searched refs:extra_dependencies (Results 1 – 6 of 6) 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()
103 const absl::Span<const TopologicalDependency> extra_dependencies, in ComputeTopologicalOrder() argument
107 ComputeTopologicalOrder(graph, extra_dependencies, &ready_nodes)); in ComputeTopologicalOrder()
|
| D | topological_sort_test.cc | 174 std::vector<TopologicalDependency> extra_dependencies; in TEST_F() local
175 extra_dependencies.push_back({&graph.node(5), &graph.node(4)}); in TEST_F()
178 TF_EXPECT_OK(ComputeTopologicalOrder(graph, extra_dependencies, &topo_order)); in TEST_F()
194 extra_dependencies.push_back({&graph.node(1), &graph.node(5)}); in TEST_F()
196 ComputeTopologicalOrder(graph, extra_dependencies, &topo_order).ok()); in TEST_F()
|
| D | topological_sort.h | 44 absl::Span<const TopologicalDependency> extra_dependencies,
|
| /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 | 95 extra_dependencies = self._extra_dependencies.get(obj, {})
99 if name in extra_dependencies:
100 yield base.TrackableReference(name, extra_dependencies[name])
103 for name, dep in extra_dependencies.items():
|