如何在 GPU 支持下編譯 tflite？

Question

我嘗試使用 GPU 支持為 Android 編譯 tflite

bazel build --cxxopt='--std=c++11' -c opt --fat_apk_cpu=arm64-v8a //tensorflow/lite/java:tensorflowlite_gpu

但我收到幾條錯誤消息，例如：

./tensorflow/lite/delegates/gpu/gl/gl_call.h:60:29: error: no viable conversion from returned value of type 'tflite::gpu::Status' to function return type 'int'
    if (status.ok()) return OkStatus();

我是否需要將任何其他命令行參數傳遞給 bazel？

Answer 1

您需要-DMESA_EGL_NO_X11_HEADERS標志。 見https://github.com/tensorflow/tensorflow/issues/27320