diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-02-06 14:54:18 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-10 20:01:04 +0000 |
commit | 5728713fca4f6e2dff60dad3689e471545e563d2 (patch) | |
tree | 848421100f82a33ff57ee3205c369ad75737f7d3 /reference_model/src/ops/op_factory.cc | |
parent | c1e25f5755997e65ac1a360ec1e875db06040d8d (diff) | |
download | reference_model-5728713fca4f6e2dff60dad3689e471545e563d2.tar.gz |
Add FFT2d to the reference model
Includes:
* FFT2d reference implementation
* Basic TOSA tests
Change-Id: Ie79fcb713542345d550ec013646810c1e890e388
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Diffstat (limited to 'reference_model/src/ops/op_factory.cc')
-rw-r--r-- | reference_model/src/ops/op_factory.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index b1a405a..8d84135 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -89,6 +89,9 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_THREE_TYPE(OpDepthwiseConv2d, INT8, INT8, INT32); DEF_FACTORY_THREE_TYPE(OpDepthwiseConv2d, INT16, INT8, INT48); break; + case Op_FFT2D: + DEF_FACTORY_ONE_TYPE(OpFFT2d, FP32); + break; case Op_FULLY_CONNECTED: DEF_FACTORY_THREE_TYPE(OpFullyConnected, FP16, FP16, FP16); DEF_FACTORY_THREE_TYPE(OpFullyConnected, FP16, FP16, FP32); |