Is Flex delegate compatible with CUDA?

Skip to first unread message

Iosif Hamlatzis

Jan 26, 2022, 2:54:27 AMJan 26
to TensorFlow Developers
I have created a model with SELECT OPS but if I have CUDA enabled it crashes both under Python and C++, but works fine if CUDA isn't enabled

When CUDA is enabled (and only then) I get an error message 

"op 'ReadVariableOp' output%d tensor#%d shape mismatch for [144] != []" from file "./tensorflow/lite/delegates/flex/" at line 590 and then it goes to file "./tensorflow/lite/delegates/flex/" in method "CopyFromBufferHandle" in line 160
and crashes "memcpy(output->data.raw,, t_data.size());"

Any idea?
Reply all
Reply to author
0 new messages