Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc480 std::unordered_map<string, TensorSizeHistogram> device_class_to_hist_map; in GetStatsStringFromRunMetadata() local
488 auto it = device_class_to_hist_map.find(device_class); in GetStatsStringFromRunMetadata()
489 if (it == device_class_to_hist_map.end()) { in GetStatsStringFromRunMetadata()
490 device_class_to_hist_map.emplace(device_class, TensorSizeHistogram(hist)); in GetStatsStringFromRunMetadata()
497 for (const auto& device_hist : device_class_to_hist_map) { in GetStatsStringFromRunMetadata()