Support kTfLiteComplex64 in GPU
6 views
Skip to first unread message
Iosif Hamlatzis
Feb 15, 2023, 4:43:27 AM2/15/23
to TensorFlow Lite
Some tflite kernels in GPU also support kTfLiteComplex64 type tensors. When porting those kernels to GPU it's easy to support various tensor's types for instance by converting them from int to float when uploading to GPU memory and back to int from downloading from GPU memory to RAM; but what about supporting kTfLiteComplex64?
kTfLiteComplex64 is a struct of two floats (one representing the real part & the other representing the imaginary part). So does anyone have any hint on how to support kTfLiteComplex64?
thnx in advance
Reply all
Reply to author
Forward
0 new messages