如何在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