Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc282 std::unordered_map<int, const DeviceStepStats*> gpu_hardware_stats; in BuildCostModel() local
290 gpu_hardware_stats.emplace(gpu_id, &device_stats); in BuildCostModel()
304 if (gpu_hardware_stats.find(gpu_id) != gpu_hardware_stats.end()) { in BuildCostModel()
305 itr.second.hardware_stats = gpu_hardware_stats.find(gpu_id)->second; in BuildCostModel()