@tqchen Hi! I have updated the code as your comments. I add a new constructor `NDArray(const TBlob &data, int dev_id, const std::function<void()>& deleter) ` for `NDArray class`, because the performance between `std::make_shared` and `std::shared_ptr::shared_ptr` is different.
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12047 ] This message was relayed via gitbox.apache.org for [email protected]
