Home
last modified time | relevance | path

Searched refs:next_nodes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_parallelization_test.py61 next_nodes = ["ParallelMap"] if should_optimize else ["Map"]
63 testing.assert_next(next_nodes)).map(function)
107next_nodes = ["ParallelMap"] if (apply_autotune is not False) else ["Map"] # pylint: disable=g-bo…
109 testing.assert_next(next_nodes)).map(lambda x: x + 2)
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc164 StringPieceSet next_nodes; in RemoveUnusedNodes() local
179 next_nodes.insert(id.first); in RemoveUnusedNodes()
183 current_nodes.swap(next_nodes); in RemoveUnusedNodes()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_model.cc651 std::vector<int> next_nodes; in MergeNodes() local
656 next_nodes.push_back(j); in MergeNodes()
661 if (next_nodes.size() != 1 || link_index != 0) { in MergeNodes()
664 auto& linkable_node = nodes[next_nodes[0]]; in MergeNodes()
671 nodes.erase(nodes.begin() + next_nodes[0]); in MergeNodes()