Searched refs:TensorSizeHistogram (Results 1 – 3 of 3) sorted by relevance
77 class TensorSizeHistogram {79 TensorSizeHistogram() : buckets_(kMaxBuckets, 0) {} in TensorSizeHistogram() function82 void Merge(const TensorSizeHistogram& src);
338 void TensorSizeHistogram::Add(const uint64 value) { in Add()346 void TensorSizeHistogram::Merge(const TensorSizeHistogram& src) { in Merge()355 string TensorSizeHistogram::ToString() const { in ToString()403 const int TensorSizeHistogram::Index(const uint64 value) const { in Index()459 std::unordered_map<string, TensorSizeHistogram> device_to_hist_map; in GetStatsStringFromRunMetadata()480 std::unordered_map<string, TensorSizeHistogram> device_class_to_hist_map; in GetStatsStringFromRunMetadata()490 device_class_to_hist_map.emplace(device_class, TensorSizeHistogram(hist)); in GetStatsStringFromRunMetadata()
228 class TestTensorSizeHistogram : public TensorSizeHistogram {