Browse Source

!10378 [lite] rollback model.h

From: @xu_anyue
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ce7b3ea9d1
3 changed files with 7 additions and 2 deletions
  1. +3
    -0
      mindspore/lite/include/model.h
  2. +1
    -1
      mindspore/lite/src/lite_model.cc
  3. +3
    -1
      mindspore/lite/src/lite_model.h

+ 3
- 0
mindspore/lite/include/model.h View File

@@ -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;
};


+ 1
- 1
mindspore/lite/src/lite_model.cc View File

@@ -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) {


+ 3
- 1
mindspore/lite/src/lite_model.h View File

@@ -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