1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/bcast_ops.cc b/tensorflow/core/kernels/bcast_ops.cc
index 7a45f4cfea..8ab038a4b7 100644
--- a/tensorflow/core/kernels/bcast_ops.cc
+++ b/tensorflow/core/kernels/bcast_ops.cc
@@ -60,10 +60,10 @@ class BCastGradArgsOp : public OpKernel {
 
  private:
   void Output(OpKernelContext* ctx, int idx, const BCast::Vec& v) {
-    const int len = v.size();
+    const int64 len = v.size();
     Tensor* o = nullptr;
     OP_REQUIRES_OK(ctx, ctx->allocate_output(idx, TensorShape({len}), &o));
-    for (int i = 0; i < len; ++i) o->flat<int32>()(i) = v[i];
+    for (int64 i = 0; i < len; ++i) o->flat<int32>()(i) = v[i];
   }
 
   TF_DISALLOW_COPY_AND_ASSIGN(BCastGradArgsOp);