46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
# Description:
|
|
# BenchmarkModel Android harness for TensorFlow Lite benchmarks.
|
|
|
|
package(default_visibility = ["//visibility:private"])
|
|
|
|
licenses(["notice"]) # Apache 2.0
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
load("//tensorflow/lite:build_def.bzl", "tflite_jni_binary")
|
|
load("@build_bazel_rules_android//android:rules.bzl", "android_binary")
|
|
|
|
# See README.md for details about building and executing this benchmark.
|
|
android_binary(
|
|
name = "benchmark_model",
|
|
srcs = glob([
|
|
"src/**/*.java",
|
|
]),
|
|
custom_package = "org.tensorflow.lite.benchmark",
|
|
manifest = "AndroidManifest.xml",
|
|
# In some platforms we don't have an Android SDK/NDK and this target
|
|
# can't be built. We need to prevent the build system from trying to
|
|
# use the target in that case.
|
|
tags = ["manual"],
|
|
deps = [":tensorflowlite_benchmark_native"],
|
|
)
|
|
|
|
tflite_jni_binary(
|
|
name = "libtensorflowlite_benchmark.so",
|
|
srcs = glob([
|
|
"jni/**/*.cc",
|
|
"jni/**/*.h",
|
|
]),
|
|
deps = [
|
|
"//tensorflow/lite/java/jni",
|
|
"//tensorflow/lite/tools/benchmark:benchmark_tflite_model_lib",
|
|
"//tensorflow/lite/tools/benchmark:logging",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "tensorflowlite_benchmark_native",
|
|
srcs = ["libtensorflowlite_benchmark.so"],
|
|
visibility = ["//visibility:private"],
|
|
)
|