diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_reshape_op.cc')
-rw-r--r-- | tensorflow/core/kernels/mkl_reshape_op.cc | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/mkl_reshape_op.cc b/tensorflow/core/kernels/mkl_reshape_op.cc index c44a6f3477..02ea9fc068 100644 --- a/tensorflow/core/kernels/mkl_reshape_op.cc +++ b/tensorflow/core/kernels/mkl_reshape_op.cc @@ -263,10 +263,7 @@ class MklReshapeOp : public OpKernel { // shape_from != shape_to), then we just copy input tensor to // output tensor with target shape (we cannot forward Mkl layout // in such case because shape has changed.) - std::vector<primitive> net; - if (dnn_data_input.CheckReorderToOpMem(output_tf_pd, output_tensor, - &net)) { - stream(stream::kind::eager).submit(net).wait(); + if (dnn_data_input.CheckReorderToOpMem(output_tf_pd, output_tensor)) { } else { OP_REQUIRES( context, output_tensor->CopyFrom(input_tensor, shape_to), |