fix reorder

apache · Aug 7, 2018 · 7f2f377 · 7f2f377
1 parent 8c797b3
commit 7f2f377
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/operator/nn/mkldnn/mkldnn_base.cc b/src/operator/nn/mkldnn/mkldnn_base.cc
@@ -476,8 +476,8 @@ void OpCheck::Init(const std::vector<mxnet::NDArray> &inputs_,
     NDArray data = inputs_[i];
     inputs.emplace_back(data.shape(), ctx, false, data.dtype());
     if (data.IsMKLDNNData() && data.IsView())
-        data = in_data[fullc::kData].Reorder2Default();
-    auto mem = inputs_[i].GetMKLDNNData();
+        data = data.Reorder2Default();
+    auto mem = data.GetMKLDNNData();
     inputs[i].CopyFrom(*mem);
   }
   for (size_t i = 0; i < outputs_.size(); i++) {