diff options
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r-- | src/TosaSerialize.cpp | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index e69fcba..8a95b68 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -1367,6 +1367,28 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::RFFT2dOp>( return tyop; } +template<> +TosaSerializationOperator * +TosaSerializationOperatorBuilder::build<mlir::tosa::FFT2dOp>( + mlir::Operation &op) const { + + bool inverse = op.getAttr("inverse").dyn_cast<mlir::BoolAttr>().getValue(); + + std::string input_real_name = GetTensorName(op.getOperand(0)); + std::string input_imag_name = GetTensorName(op.getOperand(1)); + std::string output_real_name = GetTensorName(op.getResult(0)); + std::string output_imag_name = GetTensorName(op.getResult(1)); + + TosaFFTAttribute attribute(inverse); + + TosaSerializationOperator *tyop = new TosaSerializationOperator( + Op_FFT2D, Attribute_FFTAttribute, &attribute, + std::vector<std::string>{input_real_name, input_imag_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(); |