aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/mkl_tfconv_op.h
diff options
context:
space:
mode:
authorGravatar Guozhong Zhuang <guozhong.zhuang@intel.com>2018-06-21 09:24:49 -0700
committerGravatar Guozhong Zhuang <guozhong.zhuang@intel.com>2018-06-21 09:24:49 -0700
commit437cfcfda7725f9cb373112f0e24b7bdf279b971 (patch)
treec8e91c8ea25b7a22e67e3a3815f5b8b8f7b1c940 /tensorflow/core/kernels/mkl_tfconv_op.h
parent181110371dac5939f8d009de08a65e51340d591f (diff)
implementation of primitive reuse for reorder
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));