diff --git a/mindspore/lite/src/lite_mindrt.h b/mindspore/lite/src/lite_mindrt.h index 076e006745b..3de2fc78139 100644 --- a/mindspore/lite/src/lite_mindrt.h +++ b/mindspore/lite/src/lite_mindrt.h @@ -53,6 +53,8 @@ class LiteOpActor : public OpActor { isolate_input_tensor->set_data(nullptr); delete isolate_input_tensor; } + delete call_node_; + delete partial_node_; } void RunOpData(OpData *input_data, OpContext *context = nullptr) override; virtual int CompileArrow(); @@ -109,7 +111,12 @@ class LiteOpActor : public OpActor { class LiteSwitchOpActor : public LiteOpActor { public: explicit LiteSwitchOpActor(kernel::LiteKernel *kernel) : LiteOpActor(kernel) {} - ~LiteSwitchOpActor() override = default; + ~LiteSwitchOpActor() override { + delete call_node_; + delete switch_node_; + delete true_partial_node_; + delete false_partial_node_; + }; void RunOpData(OpData *inputs, OpContext *context = nullptr) override; int CompileArrow() override; int PrepareOutputData() override;