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> {
* \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.
......
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