Searched refs:compact_shape_ (Results 1 – 1 of 1) sorted by relevance
767 absl::flat_hash_map<const HloInstruction*, Shape> compact_shape_; member in xla::__anon95ba67740111::MemoryUsageTracker1242 auto it = compact_shape_.find(hlo); in GetCompactShape()1243 if (it != compact_shape_.end()) { in GetCompactShape()1248 compact_shape_[hlo] = min_shape; in GetCompactShape()