GPU Support for tensorflow lite - Githubissues

bytedeco / javacpp-presets

The missing Java distribution of native C++ libraries

Other

2.68k stars 744 forks source link

GPU Support for tensorflow lite #1532

Open barrypitman opened 3 months ago

barrypitman commented 3 months ago

This PR includes some changes to enable building tensorflow-lite with -DTFLITE_ENABLE_GPU=ON for linux x64-64.

There are still some outstanding issues though:

for linux-x86_64 (-gpu), its building the jar file, but then failing during staging / deploy based on the tensorflow-lite/platform/gpu/pom.xml that I provided.
for android it's failing because of TfLiteGpuDelegateV2CreateAsync
for windows I think that the target functions aren't available since tflite doesn't support GPU on Windows, but I'm not sure how to conditionally exclude them