[Container] Fix NDArray SaveDLTensor declaration and implementation s…

…ignature different (#4586)
apache · Dec 26, 2019 · 373ab31 · 373ab31
1 parent a457df8
commit 373ab31
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.