diff options
author | 2018-03-26 17:02:55 -0700 | |
---|---|---|
committer | 2018-03-26 17:05:34 -0700 | |
commit | eda7aa3f7e763734f5f3550bed8b044a384b2ce8 (patch) | |
tree | 41dc456e35791bcced7731d27ca5118fff20a5be /tensorflow/core/kernels/mkl_reshape_op.cc | |
parent | 307cfe7ab7e2c475b2741fc2a2f7663b46223e6d (diff) |
Add missing parameter to OP_REQUIRES call.
PiperOrigin-RevId: 190548854
Diffstat (limited to 'tensorflow/core/kernels/mkl_reshape_op.cc')
-rw-r--r-- | tensorflow/core/kernels/mkl_reshape_op.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/mkl_reshape_op.cc b/tensorflow/core/kernels/mkl_reshape_op.cc index e12f6f437a..2cfde1f6fd 100644 --- 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 { |