commit 95b8235761baa6abc876fc4658488ab57cf64d5e
author tg-at-google <taregaskin@google.com> Tue Jul 28 20:56:10 2020 -0400
committer GitHub <noreply@github.com> Tue Jul 28 20:56:10 2020 -0400
tree 6e4e431689d4da898509504e8f708cff9db16ebe
parent cb3cf00aade8e747783b57debe324d6bc00b77b3

Update gradients.cc

tensorflow/cc/framework/gradients.cc

diff --git a/tensorflow/cc/framework/gradients.cc b/tensorflow/cc/framework/gradients.cc
index 4229c35..3195a35 100644
--- a/tensorflow/cc/framework/gradients.cc
+++ b/tensorflow/cc/framework/gradients.cc
@@ -524,9 +524,8 @@
     // make this association explicit.
     for (const Edge* e : n->in_edges()) {
       if (e->IsControlEdge()) continue;
-      int dx_index = e->dst_input();
-      const int dx_size = dx.size();
-      if (dx_index >= dx_size) {
+      size_t dx_index = e->dst_input();
+      if (dx_index >= dx.size()) {
         return errors::Internal(
             "Invalid gradient output index: ", dx_index, " size: ", dx.size());
       }