| 343 | } |
| 344 | |
| 345 | void TensorSizeHistogram::Add(const uint64 value) { |
| 346 | num_elem_++; |
| 347 | sum_elem_ += value; |
| 348 | min_ = std::min(min_, value); |
| 349 | max_ = std::max(max_, value); |
| 350 | buckets_[Index(value)]++; |
| 351 | } |
| 352 | |
| 353 | void TensorSizeHistogram::Merge(const TensorSizeHistogram& src) { |
| 354 | num_elem_ += src.num_elem_; |