vee1e/tflite-micro
1
0
Fork 0
mirror of https://github.com/vee1e/tflite-micro.git synced 2026-09-02 02:07:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add py_library generate_test_for_model to tflite-micro package (#2667)

Browse source 
This enables running bazel run
..:generate_micro_mutable_op_resolver_from_model_test when tflite_micro is installed.

BUG=partially fixing https://github.com/tensorflow/tflite-micro/issues/2564
This commit is contained in:
Måns Nilsson 2024-08-19 18:02:07 +02:00 • committed by GitHub
parent 07e7f33eb5
commit c01ca97f52
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
python/tflite_micro/BUILD
View file

@ -78,6 +78,7 @@ py_library(
visibility = ["//visibility:public"],
deps = [
requirement("numpy"),
"//tensorflow/lite/micro/tools:generate_test_for_model",
"//tensorflow/lite/tools:flatbuffer_utils",
],
)
@ -132,6 +133,7 @@ py_package(
# in the tflm tree.
packages = [
"python.tflite_micro",
"tensorflow.lite.micro.tools.generate_test_for_model",
"tensorflow.lite.python",
"tensorflow.lite.tools.flatbuffer_utils",
],

1
tensorflow/lite/micro/tools/BUILD
View file

@ -32,6 +32,7 @@ py_library(
name = "generate_test_for_model",
srcs = ["generate_test_for_model.py"],
srcs_version = "PY3",
visibility = ["//:__subpackages__"],
deps = [
"//tensorflow/lite/python:schema_py",
],