Home
last modified time | relevance | path

Searched refs:CostModelManager (Results 1 – 8 of 8) 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.h32 class CostModelManager; variable
165 CostModelManager* cost_model_manager,
Ddirect_session_with_tracking_alloc_test.cc98 CostModelManager::CostModelMap cost_models; in TEST()
159 CostModelManager::CostModelMap cost_models; in TEST()
216 CostModelManager::CostModelMap cost_models; in TestHWAccelerator()
Ddirect_session.h117 void ExportCostModels(CostModelManager::CostModelMap* cost_models) { in ExportCostModels()
422 CostModelManager cost_model_manager_;
Dstep_stats_collector.cc260 CostModelManager* cost_model_manager, in BuildCostModel()
/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc87 CostModelManager cost_model_manager; in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.h150 CostModelManager cost_model_manager_;