Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_parallelization_test.py64 next_nodes = ["ParallelMap"] if should_optimize else ["Map"]
66 testing.assert_next(next_nodes)).map(function)
110next_nodes = ["ParallelMap"] if (apply_autotune is not False) else ["Map"] # pylint: disable=g-bo…
112 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/metal/
Dinference_context.cc324 std::vector<int> next_nodes; in Merge() local
329 next_nodes.push_back(j); in Merge()
334 if (next_nodes.size() != 1 || link_index != 0) { in Merge()
337 auto& linkable_node = nodes_[next_nodes[0]]; in Merge()
349 nodes_.erase(nodes_.begin() + next_nodes[0]); in Merge()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc417 std::vector<int> next_nodes; in Merge() local
422 next_nodes.push_back(j); in Merge()
427 if (next_nodes.size() != 1 || link_index != 0) { in Merge()
430 auto& linkable_node = nodes_[next_nodes[0]]; in Merge()
444 nodes_.erase(nodes_.begin() + next_nodes[0]); in Merge()