diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_tfconv_op.h')
-rw-r--r-- | tensorflow/core/kernels/mkl_tfconv_op.h | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/mkl_tfconv_op.h b/tensorflow/core/kernels/mkl_tfconv_op.h index 7e8ed1b1d6..f4f0035f26 100644 --- a/tensorflow/core/kernels/mkl_tfconv_op.h +++ b/tensorflow/core/kernels/mkl_tfconv_op.h @@ -111,10 +111,8 @@ class MklToTfOp : public OpKernel { // Do we need to reorder Mkl layout into TensorFlow layout? if (input.IsReorderNeeded(output_tf_pd)) { // Insert reorder between Mkl layout and TensorFlow layout. - std::vector<primitive> net; - CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, output_tensor, &net), + CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, output_tensor), true); - stream(stream::kind::eager).submit(net).wait(); } else { // If not, just forward input tensor to output tensor. CHECK(output_tensor->CopyFrom(input_tensor, output_shape)); |