I want to use my Jetson with complex-valued RF data. Ideally, I’d like to be able to add operations such as the FFT into my TensorFlow graph, but this operation is not supported in TensorRT as it requires a complex dtype. Will TensorRT ever support complex dtypes?
This looks like a Jetson issue. Please refer to the below samples in case useful.
For any further assistance, we will move this post to to Jetson related forum.
This does not answer my question.
TensorRT doesn’t support complex number/FFT/IFFT currently.
We will check this with our internal team and get back to you.
Based on your requirement, maybe you can achieve this with a plugin layer.
Please wrap the complex number with two floating inputs.
And write a plugin layer with our CUDA FFT example below: