Support translation of RFFT2d

Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: Ic69e006fb9369414f4174e5bb24adadae531775d

1 files changed, 16 insertions, 0 deletions

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index 4bc67ea..547da8c 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -1351,6 +1351,22 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::WhileOp>(
   return tyop;
 }
 
+template <>
+TosaSerializationOperator *
+TosaSerializationOperatorBuilder::build<mlir::tosa::RFFT2dOp>(
+    mlir::Operation &op) const {
+  std::string input_name = GetTensorName(op.getOperand(0));
+  std::string output_real_name = GetTensorName(op.getResult(0));
+  std::string output_imag_name = GetTensorName(op.getResult(1));
+
+  TosaSerializationOperator *tyop = new TosaSerializationOperator(
+    Op_RFFT2D, Attribute_NONE, nullptr,
+    std::vector<std::string>{input_name},
+    std::vector<std::string>{output_real_name, output_imag_name});
+
+  return tyop;
+}
+
 /* End translating TOSA operator */
 mlir::LogicalResult TosaSerializationRegionBuilder::BuildAllBlocksInRegion(std::vector<mlir::Value>& return_values) {
   std::string region_name = ser_region->GetName();