tflite-micro/tensorflow/lite/python/BUILD
RJ Ascani 29526e413b
Replace tensorflow-cpu usage with tensorflow (#2396)
The tensorflow-cpu package does not support MacOS or non-x86 hardware. Replacing the tensorflow-cpu python package requirement with the tensorflow meta package should enable the bazel build and the dependent python scripts to be used on those platforms.

BUG=#2367, #1781
2024-02-23 00:35:22 +00:00

25 lines
556 B
Text

load("@flatbuffers//:build_defs.bzl", "flatbuffer_py_library")
load("@tflm_pip_deps//:requirements.bzl", "requirement")
package(
default_visibility = [
"//visibility:public",
],
licenses = ["notice"],
)
flatbuffer_py_library(
name = "schema_py",
srcs = ["//tensorflow/lite/schema:schema.fbs"],
)
py_library(
name = "schema_util",
srcs = ["schema_util.py"],
srcs_version = "PY3",
visibility = ["//:__subpackages__"],
deps = [
requirement("flatbuffers"),
requirement("tensorflow"),
],
)