commit 78d0c316703f4d09a1c6f111b1b604df474a6374
author Samuel Marks <807580+SamuelMarks@users.noreply.github.com> Tue Oct 19 11:07:59 2021 -0400
committer Samuel Marks <807580+SamuelMarks@users.noreply.github.com> Tue Oct 19 11:07:59 2021 -0400
tree 88c11395c12de9bc06dc1072a54159066c27f078
parent cf690d9dc6fd8ac103246168b815e91c3311cd8c

[tensorflow/compiler/mlir/xla/transforms/adjust_layout.cc] Reserve SmallVector allocation correctly

tensorflow/compiler/mlir/xla/transforms/adjust_layout.cc

diff --git a/tensorflow/compiler/mlir/xla/transforms/adjust_layout.cc b/tensorflow/compiler/mlir/xla/transforms/adjust_layout.cc
index 1ef7567..f64079d 100644
--- a/tensorflow/compiler/mlir/xla/transforms/adjust_layout.cc
+++ b/tensorflow/compiler/mlir/xla/transforms/adjust_layout.cc
@@ -78,7 +78,8 @@
     if (type.isa<TupleType>()) {
       auto tuple_type = type.dyn_cast<TupleType>();
       auto types = tuple_type.getTypes();
-      llvm::SmallVector<mlir::Attribute, types.size()> v;
+      llvm::SmallVector<mlir::Attribute> v;
+      v.reserve(types.size());
       for (const mlir::Type &t : types) {
         auto layout = GetLayout(t, rewriter);
         if (failed(layout)) return failure();