From a68274931b24625f5576ca9ee9d4268cd80fc7c8 Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Wed, 16 Nov 2022 10:41:55 -0800
Subject: Fix reference model memory leaks for the following ops

- OpClamp
- OpArithmeticRightShift
- OpMul
- OpTable
- OpTranspose

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Icb84a8a17c298b471a635310454775977a9133cb
---
 reference_model/src/ops/data_layout.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'reference_model/src/ops/data_layout.cc')

diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc
index bffd659..b127cba 100644
--- a/reference_model/src/ops/data_layout.cc
+++ b/reference_model/src/ops/data_layout.cc
@@ -581,7 +581,9 @@ OpTranspose<Rank, Dtype>::OpTranspose(SubgraphTraverser* sgt_,
 
 template <int Rank, DType Dtype>
 OpTranspose<Rank, Dtype>::~OpTranspose()
-{}
+{
+    if (attribute) delete attribute;
+}
 
 template <int Rank, DType Dtype>
 int OpTranspose<Rank, Dtype>::checkTensorAttributes()
-- 
cgit v1.2.1