From 20eeb11ee0817ebe3ba7621f5f9c29c6e43786c7 Mon Sep 17 00:00:00 2001 From: tg-at-google Date: Tue, 28 Jul 2020 16:17:29 -0400 Subject: [PATCH] Update tensorflow/compiler/mlir/tensorflow/translate/import_model.cc Co-authored-by: Mihai Maruseac --- tensorflow/compiler/mlir/tensorflow/translate/import_model.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc index 2bf2c900cd2..63366911f63 100644 --- a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc +++ b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc @@ -2387,8 +2387,8 @@ GraphDefImporter::GetArgsRetsAndTypesFromFunctionGraph( "' is missing attribute 'index'"); auto index = attr->i(); - const int nodes_size = nodes->size(); - if (nodes_size < index + 1) nodes->resize(index + 1); + const int num_nodes = nodes->size(); + if (num_nodes < index + 1) nodes->resize(index + 1); if ((*nodes)[index].node != nullptr) return errors::InvalidArgument(node->type_string(), " node '",