[Container] Fix NDArray SaveDLTensor declaration and implementation s…

…ignature different (apache#4586)
zhiics · Dec 31, 2019 · 4bb799d · 4bb799d
1 parent 24275a6
commit 4bb799d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/tvm/runtime/ndarray.h b/include/tvm/runtime/ndarray.h
@@ -228,7 +228,7 @@ struct array_type_info<NDArray> {
  * \param strm The outpu stream
  * \param tensor The tensor to be saved.
  */
-inline bool SaveDLTensor(dmlc::Stream* strm, const DLTensor* tensor);
+inline bool SaveDLTensor(dmlc::Stream* strm, DLTensor* tensor);
 
 /*!
  * \brief Reference counted Container object used to back NDArray.