include delegate headers

include GPU and Hexagon delegate headers explicitly
2020-04-02 11:53:44 +08:00 · 2020-04-02 11:53:44 +08:00 · 1e00a5a770
commit 1e00a5a770
parent 347fef645b
1 changed files with 7 additions and 0 deletions
--- a/tensorflow/lite/examples/label_image/label_image.cc
+++ b/tensorflow/lite/examples/label_image/label_image.cc
@ -45,6 +45,13 @@ limitations under the License.
 #include "tensorflow/lite/string_util.h"
 #include "tensorflow/lite/tools/evaluation/utils.h"

+#if defined(__ANDROID__)
+#include "tensorflow/lite/delegates/gpu/delegate.h"
+#if (defined(__arm__) || defined(__aarch64__))
+#include "tensorflow/lite/experimental/delegates/hexagon/hexagon_delegate.h"
+#endif
+#endif
+
 #define LOG(x) std::cerr

 namespace tflite {