Merge pull request #52864 from tensorflow/fix-build-1-on-r2.6

Fix build broken by merge conflict in `tensorflow/core/kernels/sequen…
tensorflow · Oct 29, 2021 · c42e447 · c42e447
2 parents 0e258b8 + fd11c17
commit c42e447
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/sequence_ops.cc b/tensorflow/core/kernels/sequence_ops.cc
@@ -78,6 +78,8 @@ class RangeOp : public OpKernel {
     } else {
       size = static_cast<int64>(std::ceil(std::abs((limit - start) / delta)));
     }
+    TensorShape shape;
+    OP_REQUIRES_OK(context, shape.AddDimWithStatus(size));
     Tensor* out = nullptr;
     OP_REQUIRES_OK(context, context->allocate_output(0, shape, &out));
     auto flat = out->flat<T>();