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.cc99 CostModelManager::CostModelMap cost_models; in TEST() local
100 ds->ExportCostModels(&cost_models); in TEST()
101 for (auto& it : cost_models) { in TEST()
160 CostModelManager::CostModelMap cost_models; in TEST() local
161 ds->ExportCostModels(&cost_models); in TEST()
162 CHECK_GE(cost_models.size(), 1); in TEST()
163 const CostModel* cm = (*cost_models.begin()).second; in TEST()
218 CostModelManager::CostModelMap cost_models; in TestHWAccelerator() local
219 ds->ExportCostModels(&cost_models); in TestHWAccelerator()
220 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.h117 void ExportCostModels(CostModelManager::CostModelMap* cost_models) { in ExportCostModels() argument
118 cost_model_manager_.ExportCostModels(cost_models); in ExportCostModels()