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

您需要value标志。参见https://github.com/tensorflow/tensorflow/issues/27320