Commit 373ab314 by Zhao Wu Committed by Tianqi Chen

[Container] Fix NDArray SaveDLTensor declaration and implementation signature different (#4586)

parent a457df8a
...@@ -228,7 +228,7 @@ struct array_type_info<NDArray> { ...@@ -228,7 +228,7 @@ struct array_type_info<NDArray> {
* \param strm The outpu stream * \param strm The outpu stream
* \param tensor The tensor to be saved. * \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. * \brief Reference counted Container object used to back NDArray.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment