!10378 [lite] rollback model.h

From: @xu_anyue
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
pull/10378/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit ce7b3ea9d1

@ -55,6 +55,9 @@ struct MS_API Model {
/// \brief Free meta graph temporary buffer
virtual void Free() = 0;
/// \brief Free all temporay buffer.EG: nodes in the model.
virtual void Destroy() = 0;
/// \brief Model destruct, free all memory
virtual ~Model() = default;
};

@ -101,7 +101,7 @@ void LiteModel::Free() {
attr_tensor_bufs_.resize(0);
}
LiteModel::~LiteModel() {
void LiteModel::Destroy() {
Free();
auto nodes_size = this->all_nodes_.size();
for (size_t i = 0; i < nodes_size; ++i) {

@ -42,7 +42,9 @@ class LiteModel : public Model {
void Free() override;
~LiteModel() override;
void Destroy() override;
~LiteModel() override { Destroy(); }
private:
#ifdef ENABLE_V0

Loading…
Cancel
Save