Home
last modified time | relevance | path

Searched refs:cost_models (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc98 CostModelManager::CostModelMap cost_models; in TEST() local
99 ds->ExportCostModels(&cost_models); in TEST()
100 for (auto& it : cost_models) { in TEST()
159 CostModelManager::CostModelMap cost_models; in TEST() local
160 ds->ExportCostModels(&cost_models); in TEST()
161 CHECK_GE(cost_models.size(), 1); in TEST()
162 const CostModel* cm = (*cost_models.begin()).second; in TEST()
225 CostModelManager::CostModelMap cost_models; in TestHWAccelerator() local
226 ds->ExportCostModels(&cost_models); in TestHWAccelerator()
227 for (auto& it : cost_models) { in TestHWAccelerator()
Dcostmodel_manager.h37 void ExportCostModels(CostModelMap* cost_models) { in ExportCostModels() argument
39 *cost_models = cost_models_; in ExportCostModels()
Ddirect_session.h107 void ExportCostModels(CostModelManager::CostModelMap* cost_models) { in ExportCostModels() argument
108 cost_model_manager_.ExportCostModels(cost_models); in ExportCostModels()