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