Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_parallelization_test.py56 next_nodes = ["ParallelMap"] if should_be_parallel else ["Map"]
58 optimization.assert_next(next_nodes)).map(function)
/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()