From 1182c93e3ea447477554a6825014199f455d86d4 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 5 May 2017 22:46:14 -0800
Subject: [PATCH] Fix warning of type conversion in array_ops.cc Change:
155273711
---
tensorflow/core/ops/array_ops.cc | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tensorflow/core/ops/array_ops.cc b/tensorflow/core/ops/array_ops.cc
index 5e5ede8aa92..9a8b70acf11 100644
--- a/tensorflow/core/ops/array_ops.cc
+++ b/tensorflow/core/ops/array_ops.cc
@@ -550,7 +550,7 @@ REGISTER_OP("SplitV")
return errors::InvalidArgument(
"Length of size_splits should be equal to num_outputs");
}
- int cumsum_outputs = 0;
+ int64_t cumsum_outputs = 0;
bool has_neg_one = false;
// If the sizes of the splits are known, then
// make sure that the sizes add up to the expected
@@ -568,7 +568,7 @@ REGISTER_OP("SplitV")
else
cumsum_outputs += data[i];
}
- int split_dim_size = c->Value(c->Dim(input, split_dim));
+ auto split_dim_size = c->Value(c->Dim(input, split_dim));
if (has_neg_one) {
if (cumsum_outputs < split_dim_size)
cumsum_outputs = split_dim_size;