From: hlu1 <14827759+hlu1@users.noreply.github.com>
Date: Fri, 10 Apr 2020 14:42:54 +0000 (-0700)
Subject: [NDArray] Set shape_ in NDArray::FromDLPack (#5301)
X-Git-Tag: upstream/0.7.0~944
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=48082358879b5402d273715418f8d629d689af54;p=platform%2Fupstream%2Ftvm.git

[NDArray] Set shape_ in NDArray::FromDLPack (#5301)
---

diff --git a/src/runtime/ndarray.cc b/src/runtime/ndarray.cc
index 99594ee..ac12472 100644
--- a/src/runtime/ndarray.cc
+++ b/src/runtime/ndarray.cc
@@ -208,6 +208,10 @@ NDArray NDArray::FromDLPack(DLManagedTensor* tensor) {
   // fill up content.
   data->manager_ctx = tensor;
   data->dl_tensor = tensor->dl_tensor;
+  // update shape_
+  data->shape_.resize(data->dl_tensor.ndim);
+  data->shape_.assign(data->dl_tensor.shape, data->dl_tensor.shape + data->dl_tensor.ndim);
+  data->dl_tensor.shape = data->shape_.data();
   return NDArray(GetObjectPtr<Object>(data));
 }