Searched refs:TensorSizeHistogram (Results 1 – 3 of 3) sorted by relevance
78 class TensorSizeHistogram {80 TensorSizeHistogram() : buckets_(kMaxBuckets, 0) {} in TensorSizeHistogram() function83 void Merge(const TensorSizeHistogram& src);
339 void TensorSizeHistogram::Add(const uint64 value) { in Add()347 void TensorSizeHistogram::Merge(const TensorSizeHistogram& src) { in Merge()356 string TensorSizeHistogram::ToString() const { in ToString()386 const int TensorSizeHistogram::Index(const uint64 value) const { in Index()442 std::unordered_map<string, TensorSizeHistogram> device_to_hist_map; in GetStatsStringFromRunMetadata()463 std::unordered_map<string, TensorSizeHistogram> device_class_to_hist_map; in GetStatsStringFromRunMetadata()473 device_class_to_hist_map.emplace(device_class, TensorSizeHistogram(hist)); in GetStatsStringFromRunMetadata()
228 class TestTensorSizeHistogram : public TensorSizeHistogram {