From 2e94b05c27fb5ca7a93d997019dbcd9f92249269 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 27 Jan 2021 10:17:28 -0800
Subject: [PATCH] Pre-allocate memory for vectors where size is already known.

PiperOrigin-RevId: 354117982
Change-Id: I868507ccb54c700973dc2c923fc2dda9b3d5bfac
---
 tensorflow/python/grappler/cluster_wrapper.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/python/grappler/cluster_wrapper.cc b/tensorflow/python/grappler/cluster_wrapper.cc
index dee8e593307..59ce706140f 100644
--- a/tensorflow/python/grappler/cluster_wrapper.cc
+++ b/tensorflow/python/grappler/cluster_wrapper.cc
@@ -146,6 +146,7 @@ PYBIND11_MODULE(_pywrap_tf_cluster, m) {
     std::vector<tensorflow::OpDef> ops;
     registry->GetRegisteredOps(&ops);
     std::vector<std::string> op_names;
+    op_names.reserve(ops.size());
     for (const tensorflow::OpDef& op : ops) {
       op_names.push_back(op.name());
     }