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.cc524 std::unordered_map<string, const Graph*> device_to_graph; in BuildCostModel() local
527 device_to_graph[unit.device->name()] = unit.graph; in BuildCostModel()
530 collector->BuildCostModel(&cost_model_manager_, device_to_graph); in BuildCostModel()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc694 std::unordered_map<string, const Graph*> device_to_graph; in RunInternal() local
699 device_to_graph[device] = graph; in RunInternal()
703 run_state.collector->BuildCostModel(&cost_model_manager_, device_to_graph); in RunInternal()