diff options
Diffstat (limited to 'reference_model/src/ops')
| -rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 7 | ||||
| -rw-r--r-- | reference_model/src/ops/tensor_ops.h | 1 |
2 files changed, 7 insertions, 1 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 2d54d8e..3f0e7b2 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -1736,11 +1736,16 @@ OpRFFT2d<Dtype>::OpRFFT2d(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_ { setRequiredOperands(1, 2); setRequiredRank(3, 3); + + INIT_ATTRIBUTE(RFFT); } template <TOSA_REF_TYPE Dtype> OpRFFT2d<Dtype>::~OpRFFT2d() -{} +{ + if (attribute) + delete attribute; +} template <TOSA_REF_TYPE Dtype> int OpRFFT2d<Dtype>::checkTensorAttributes() diff --git a/reference_model/src/ops/tensor_ops.h b/reference_model/src/ops/tensor_ops.h index f5fcd7f..e2bb811 100644 --- a/reference_model/src/ops/tensor_ops.h +++ b/reference_model/src/ops/tensor_ops.h @@ -291,6 +291,7 @@ protected: TosaReference::TensorTemplate<TIn>* in; TosaReference::TensorTemplate<TOut>* out_real; TosaReference::TensorTemplate<TOut>* out_imag; + tosa::TosaRFFTAttribute* attribute; }; template <TOSA_REF_TYPE InDtype, TOSA_REF_TYPE WeightDtype, TOSA_REF_TYPE OutDtype> |