Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dgraph_analyzer.h40 GraphAnalyzer(const GraphDef& graph, int subgraph_size);
Dgraph_analyzer.cc29 GraphAnalyzer::GraphAnalyzer(const GraphDef& graph, int subgraph_size) in GraphAnalyzer() argument
30 : graph_(graph), subgraph_size_(subgraph_size) {} in GraphAnalyzer()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc914 auto subgraph_size = subgraphs.size(); in InstantiateMultiDevice() local
915 gtl::InlinedVector<Status, 4> instantiate_status(subgraph_size); in InstantiateMultiDevice()
916 BlockingCounter counter(static_cast<int>(subgraph_size)); in InstantiateMultiDevice()
917 auto runner = [this, subgraph_size](std::function<void()> fn) { in InstantiateMultiDevice()
921 if (default_thread_pool_ != nullptr && subgraph_size > 8) { in InstantiateMultiDevice()