diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-01-10 14:50:31 +0000 |
---|---|---|
committer | Luke Hutton <luke.hutton@arm.com> | 2023-01-24 13:40:17 +0000 |
commit | 261b7b62b959a6c7312d810d9152069fdff69f3e (patch) | |
tree | 2be25cefa14cd21379a9fc6f6c499622b6de8bf8 /reference_model/src/ops/op_factory.cc | |
parent | c253e64710f22016894c0e3ac4e9eb76d62cb2f9 (diff) | |
download | reference_model-261b7b62b959a6c7312d810d9152069fdff69f3e.tar.gz |
Add RFFT2d to the reference model
Includes:
* RFFT2d reference implementation
* TFLite framework tests
* Basic TOSA tests
* Serialization submodule upgrade with support for FFT/RFFT
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I2a687e9cf87fb62a26160ea52439ba9830bea36e
Diffstat (limited to 'reference_model/src/ops/op_factory.cc')
-rw-r--r-- | reference_model/src/ops/op_factory.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index 0121ccf..0d56161 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2022, ARM Limited. +// Copyright (c) 2020-2023, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -113,6 +113,9 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_ONE_TYPE(OpMaxPool2d, INT8); DEF_FACTORY_ONE_TYPE(OpMaxPool2d, INT16); break; + case Op_RFFT2D: + DEF_FACTORY_ONE_TYPE(OpRFFT2d, FP32); + break; case Op_TRANSPOSE_CONV2D: DEF_FACTORY_TWO_TYPE_ONE_ACCUM(OpTransposeConv2d, TransposeConv, FP16, FP16, FP16); DEF_FACTORY_TWO_TYPE_ONE_ACCUM(OpTransposeConv2d, TransposeConv, FP16, FP16, FP32); |