Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dmodel.cc2803 auto bfs_nodes = CollectNodes(root_, TraversalOrder::BFS, IsAnyNode); in ModelTiming() local
2804 auto reverse_bfs_nodes = bfs_nodes; in ModelTiming()
2806 ComputePipelineRatios(bfs_nodes); in ModelTiming()
2835 void ModelTiming::ComputePipelineRatios(const Node::NodeVector& bfs_nodes) { in ComputePipelineRatios() argument
2836 for (const auto& node : bfs_nodes) { in ComputePipelineRatios()
2986 auto bfs_nodes = CollectNodes(root_, TraversalOrder::BFS, IsAnyNode); in GetStageRoots() local
2988 if (!bfs_nodes.empty() && !bfs_nodes[0]->IsAsync()) { in GetStageRoots()
2989 roots.push_back(bfs_nodes[0]); in GetStageRoots()
2991 for (auto& node : bfs_nodes) { in GetStageRoots()
Dmodel.h1025 void ComputePipelineRatios(const Node::NodeVector& bfs_nodes);