| /external/tensorflow/tensorflow/core/grappler/verifiers/ |
| structure_verifier.cc | 43 std::vector<const NodeDef*> topo_order; local
44 status_group.Update(ComputeTopologicalOrder(graph, &topo_order));
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| topological_sort_test.cc | 68 std::vector<const NodeDef*> topo_order; local
69 TF_EXPECT_OK(ComputeTopologicalOrder(graph, &topo_order));
73 ASSERT_EQ(topo_order.size(), order.size());
74 for (int i = 0; i < topo_order.size(); ++i) {
75 const NodeDef* node = topo_order[i];
80 for (int i = 0; i < topo_order.size(); i++) {
95 std::vector<const NodeDef*> topo_order; local
96 TF_EXPECT_OK(ComputeTopologicalOrder(graph, &topo_order));
100 ASSERT_EQ(topo_order.size(), order.size());
101 for (int i = 0; i < topo_order.size(); ++i)
177 std::vector<const NodeDef*> topo_order; local
205 std::vector<const NodeDef*> topo_order; local
[all...] |
| topological_sort.h | 36 // topological order to the `topo_order` output argument.
45 std::vector<const NodeDef*>* topo_order);
47 std::vector<const NodeDef*>* topo_order);
|
| topological_sort.cc | 104 std::vector<const NodeDef*>* topo_order) {
109 topo_order->reserve(ready_nodes.size());
111 topo_order->emplace_back(&graph.node(ready_node_idx));
118 std::vector<const NodeDef*>* topo_order) {
119 return ComputeTopologicalOrder(graph, {}, topo_order);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 423 explicit TopoQueue(const std::vector<const NodeDef*>& topo_order)
424 : topo_order_(TopoOrder(topo_order)) {}
450 const std::vector<const NodeDef*>& topo_order) const {
452 map.reserve(topo_order.size());
453 for (int i = 0; i < topo_order.size(); ++i) {
454 map.emplace(topo_order[i], i);
2110 std::vector<const NodeDef*> topo_order; local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| memory_optimizer.cc | 593 std::unordered_map<const NodeDef*, int> topo_order; local
595 DfsCallbacks::PostOrder([&topo_order](const NodeDef* n) {
596 int topo_index = static_cast<int>(topo_order.size());
597 topo_order[n] = topo_index;
606 input_topo_index.push_back(topo_order.at(node));
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| hierarchical_controller.py | 433 topo_order = [op.name for op in self.important_ops]
437 for idx, x in enumerate(topo_order):
507 for idx, op_name in enumerate(topo_order):
[all...] |
