aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/mkl_tfconv_op.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/mkl_tfconv_op.h')
-rw-r--r--tensorflow/core/kernels/mkl_tfconv_op.h4
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));