Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc565 std::unordered_map<string, const Graph*> device_to_graph; in BuildCostModel() local
568 device_to_graph[unit.device->name()] = unit.graph.get(); in BuildCostModel()
571 collector->BuildCostModel(&cost_model_manager_, device_to_graph); in BuildCostModel()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc779 std::unordered_map<string, const Graph*> device_to_graph; in RunInternal() local
784 device_to_graph[device] = graph; in RunInternal()
788 run_state.collector->BuildCostModel(&cost_model_manager_, device_to_graph); in RunInternal()