Replaced std::vector with absl::InlinedVector.

PiperOrigin-RevId: 587136655
47-studio-org · Dec 1, 2023 · 79c1c93 · 79c1c93
1 parent 3987ee3
commit 79c1c93
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/third_party/xla/xla/layout_util.cc b/third_party/xla/xla/layout_util.cc
@@ -252,7 +252,8 @@ Layout CreateDefaultLayoutForRank(int64_t rank) {
         absl::StrJoin(layout.minor_to_major(), ", "), shape.ShortDebugString());
   }
 
-  std::vector<bool> dimensions_in_layout(shape.rank(), false);
+  absl::InlinedVector<bool, InlineRank()> dimensions_in_layout(shape.rank(),
+                                                               false);
   for (int64_t i = 0; i < shape.rank(); ++i) {
     int64_t dim = layout.minor_to_major(i);
     if (dim < 0 || dim >= shape.rank()) {