Add missing parameter to OP_REQUIRES call.

PiperOrigin-RevId: 190548854

diff --git a/tensorflow/core/kernels/mkl_reshape_op.cc b/tensorflow/core/kernels/mkl_reshape_op.cc
index e12f6f4..2cfde1f 100644 (file)
--- a/tensorflow/core/kernels/mkl_reshape_op.cc
+++ b/tensorflow/core/kernels/mkl_reshape_op.cc
@@ -266,8 +266,9 @@ class MklReshapeOp : public OpKernel {
                                                    &net)) {
               stream(stream::kind::eager).submit(net).wait();
             } else {
-              OP_REQUIRES(context,
-                          output_tensor->CopyFrom(input_tensor, shape_to));
+              OP_REQUIRES(
+                  context, output_tensor->CopyFrom(input_tensor, shape_to),
+                  errors::InvalidArgument("invalid input tensor shape"));
             }
             return;
           } else {