[mlir][tosa] Add FFT2d operation
authorLuke Hutton <luke.hutton@arm.com>
Tue, 14 Mar 2023 18:55:39 +0000 (18:55 +0000)
committerRobert Suderman <suderman@google.com>
Tue, 14 Mar 2023 19:04:52 +0000 (19:04 +0000)
commit312938864e676c6a0ba9ba173d3ded910e5e48c9
treeb73014722651b19a2d2bddfb8342fdb9e8b199d5
parentf242291f59bf5cced08e8f6e7c3068cfe4dff06c
[mlir][tosa] Add FFT2d operation

Adds the FFT2d TOSA operation and supporting
shape inference function.

Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Reviewed By: rsuderman, eric-k256

Differential Revision: https://reviews.llvm.org/D144784
mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
mlir/test/Dialect/Tosa/ops.mlir
mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir