Home
last modified time | relevance | path

Searched refs:CostModelManager (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcostmodel_manager.cc27 CostModelManager::~CostModelManager() { in ~CostModelManager()
33 CostModel* CostModelManager::FindOrCreateCostModel(const Graph* graph) { in FindOrCreateCostModel()
45 bool CostModelManager::RemoveCostModelForGraph(const Graph* graph) { in RemoveCostModelForGraph()
56 Status CostModelManager::AddToCostGraphDef(const Graph* graph, in AddToCostGraphDef()
Dcostmodel_manager.h30 class CostModelManager {
32 ~CostModelManager();
Dstep_stats_collector.h33 class CostModelManager; variable
170 CostModelManager* cost_model_manager,
Ddirect_session_with_tracking_alloc_test.cc98 CostModelManager::CostModelMap cost_models; in TEST()
159 CostModelManager::CostModelMap cost_models; in TEST()
225 CostModelManager::CostModelMap cost_models; in TestHWAccelerator()
Ddirect_session.h107 void ExportCostModels(CostModelManager::CostModelMap* cost_models) { in ExportCostModels()
400 CostModelManager cost_model_manager_;
Dstep_stats_collector.cc262 CostModelManager* cost_model_manager, in BuildCostModel()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.h149 CostModelManager cost_model_manager_;