diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_aggregate_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/mkl_aggregate_ops.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/mkl_aggregate_ops.cc b/tensorflow/core/kernels/mkl_aggregate_ops.cc index 4ad858e4a9..3d04aeeb3e 100644 --- a/tensorflow/core/kernels/mkl_aggregate_ops.cc +++ b/tensorflow/core/kernels/mkl_aggregate_ops.cc @@ -445,11 +445,10 @@ class MklAddNOp : public OpKernel { // atleast one input is in MKL format, we choose output descriptor for // reorder. std::vector<primitive::at> inputs; - std::vector<primitive> net; // Check if actual input format of the tensor is different than common_pd // we told MKLDNN. In that case, we will need reorder. - src1.CheckReorderToOpMem(srcs_pd[0], &net); - src2.CheckReorderToOpMem(srcs_pd[1], &net); + src1.CheckReorderToOpMem(srcs_pd[0]); + src2.CheckReorderToOpMem(srcs_pd[1]); inputs.push_back(src1.GetOpMem()); inputs.push_back(src2.GetOpMem()); @@ -482,6 +481,7 @@ class MklAddNOp : public OpKernel { dst.SetUsrMemDataHandle(dst_tensor); // Create Sum op, and submit net for execution. + std::vector<primitive> net; net.push_back(sum(sum_pd, inputs, dst.GetOpMem())); stream(stream::kind::eager).submit(net).wait(); } catch (mkldnn::error& e) { |