diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_reshape_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/mkl_reshape_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/mkl_reshape_op.cc b/tensorflow/core/kernels/mkl_reshape_op.cc index 753a8b52b4..593aa3a2fd 100644 --- a/tensorflow/core/kernels/mkl_reshape_op.cc +++ b/tensorflow/core/kernels/mkl_reshape_op.cc @@ -129,7 +129,7 @@ class MklReshapeOp : public OpKernel { return; } } else { - CopyTFTensorInToOut(context, 0, 0, shape); + CopyTfTensorInToOutWithShape(context, 0, 0, shape); } } }; |