diff options
| author | Tai Ly <tai.ly@arm.com> | 2023-11-13 20:18:14 +0000 |
|---|---|---|
| committer | Tai Ly <tai.ly@arm.com> | 2023-11-14 23:10:30 +0000 |
| commit | fd8fde80452ba68a21de4de53517ebc4b4aac9ea (patch) | |
| tree | 42b46a8b185ac984a12f5e834f15d909439dc812 /reference_model/src/ops | |
| parent | aee62afba99a74f772b97356fd4c18f3fdf37073 (diff) | |
| download | reference_model-fd8fde80452ba68a21de4de53517ebc4b4aac9ea.tar.gz | |
[reference_model] Add local_bound support
Add support for local_bound attributes.
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Ie1acb65ca2495fb7d1512bf120568c695635d631
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> |