如何编译支持 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

如何编译支持 GPU 的 tflite？

How to compile tflite with GPU support?

c++

java

delegates

gpu

tensorflow-lite