experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #37893 from freedomtan:update_label_image_gl_delegate_dep

Browse Source 
PiperOrigin-RevId: 305186472
Change-Id: Ia7cd309ffa4b5008f49533297d1c453af03fb835
This commit is contained in:
TensorFlower Gardener 2020-04-06 22:25:59 -07:00
commit aae07a2470
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/lite/examples/label_image/BUILD
View File

@ -40,11 +40,11 @@ cc_binary(
"@com_google_absl//absl/memory",
] + select({
"//tensorflow:android": [
"//tensorflow/lite/delegates/gpu:gl_delegate",
"//tensorflow/lite/delegates/gpu:delegate",
"//tensorflow/lite/experimental/delegates/hexagon:hexagon_delegate",
],
"//tensorflow:android_arm64": [
"//tensorflow/lite/delegates/gpu:gl_delegate",
"//tensorflow/lite/delegates/gpu:delegate",
"//tensorflow/lite/experimental/delegates/hexagon:hexagon_delegate",
],
"//conditions:default": [],
@ -68,11 +68,11 @@ cc_library(
"//tensorflow/lite/schema:schema_fbs",
] + select({
"//tensorflow:android": [
"//tensorflow/lite/delegates/gpu:gl_delegate",
"//tensorflow/lite/delegates/gpu:delegate",
"//tensorflow/lite/experimental/delegates/hexagon:hexagon_delegate",
],
"//tensorflow:android_arm64": [
"//tensorflow/lite/delegates/gpu:gl_delegate",
"//tensorflow/lite/delegates/gpu:delegate",
"//tensorflow/lite/experimental/delegates/hexagon:hexagon_delegate",
],
"//conditions:default": [],

3
tensorflow/lite/examples/label_image/label_image.cc
View File

@ -36,6 +36,9 @@ limitations under the License.
#include <vector>
#include "absl/memory/memory.h"
#if defined(__ANDROID__)
#include "tensorflow/lite/delegates/gpu/delegate.h"
#endif
#include "tensorflow/lite/delegates/nnapi/nnapi_delegate.h"
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
#include "tensorflow/lite/examples/label_image/bitmap_helpers.h"