FFT operator in Relay IR

Questions
#1

Does Relay have an FFT operator?

#2

Found a way to add it to the operator registry, that works.

1 Like
#3

Can you show me a simple code sample of how it was implemented?

#4

https://tvm.apache.org/docs/dev/relay_add_op.html, you can modify this for your convenience.

1 Like
#5

Thank you for reply!

#6

Hi,

did you implement the FFT as a C /Python library and registered to tvm? or did you also define the FFT at the TE/TIR level and used TVM’s codegen to generate the implementation?