[Container] Fix NDArray SaveDLTensor declaration and implementation s…

…ignature different (apache#4586)
neo-ai · Jan 11, 2020 · 13652ac · 13652ac
1 parent 8afeaf6
commit 13652ac
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.