Home
last modified time | relevance | path

Searched refs:cost_model_manager_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc75 graph_mgr->cost_model_manager_.RemoveCostModelForGraph(unit.graph); in ~Item()
530 collector->BuildCostModel(&cost_model_manager_, device_to_graph); in BuildCostModel()
534 cost_model_manager_.AddToCostGraphDef(unit.graph, cost_graph) in BuildCostModel()
Dgraph_mgr.h149 CostModelManager cost_model_manager_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.h108 cost_model_manager_.ExportCostModels(cost_models); in ExportCostModels()
400 CostModelManager cost_model_manager_; variable
Ddirect_session.cc703 run_state.collector->BuildCostModel(&cost_model_manager_, device_to_graph); in RunInternal()
709 cost_model_manager_.AddToCostGraphDef(item.graph, cost_graph)); in RunInternal()