78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
package(default_visibility = ["//visibility:public"])
|
|
|
|
licenses(["notice"]) # Apache 2.0
|
|
|
|
load("//tensorflow:tensorflow.bzl", "tf_py_wrap_cc")
|
|
load("//tensorflow:tensorflow.bzl", "tf_py_test")
|
|
|
|
cc_library(
|
|
name = "toco_python_api",
|
|
srcs = ["toco_python_api.cc"],
|
|
hdrs = ["toco_python_api.h"],
|
|
deps = [
|
|
"//tensorflow/contrib/lite/toco:model_flags_proto_cc",
|
|
"//tensorflow/contrib/lite/toco:toco_flags_proto_cc",
|
|
"//tensorflow/contrib/lite/toco:toco_port",
|
|
"//tensorflow/contrib/lite/toco:toco_tooling",
|
|
"//tensorflow/core:lib",
|
|
"//util/python:python_headers",
|
|
],
|
|
)
|
|
|
|
tf_py_wrap_cc(
|
|
name = "tensorflow_wrap_toco",
|
|
srcs = ["toco.i"],
|
|
deps = [
|
|
":toco_python_api",
|
|
"//tensorflow/contrib/lite/toco:model_flags_proto_cc",
|
|
"//tensorflow/contrib/lite/toco:toco_flags_proto_cc",
|
|
"//util/python:python_headers",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
py_binary(
|
|
name = "toco_from_protos",
|
|
srcs = ["toco_from_protos.py"],
|
|
srcs_version = "PY2AND3",
|
|
deps = [
|
|
":tensorflow_wrap_toco",
|
|
"//tensorflow/python:platform",
|
|
],
|
|
)
|
|
|
|
py_binary(
|
|
name = "toco_wrapper",
|
|
srcs = ["toco_wrapper.py"],
|
|
srcs_version = "PY2AND3",
|
|
deps = [
|
|
"//tensorflow:tensorflow_py",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "toco_from_protos_test",
|
|
srcs = ["toco_from_protos_test.py"],
|
|
additional_deps = [
|
|
"//tensorflow:tensorflow_py",
|
|
"//tensorflow/contrib/lite/toco:model_flags_proto_py",
|
|
"//tensorflow/contrib/lite/toco:toco_flags_proto_py",
|
|
],
|
|
data = [
|
|
":toco_from_protos",
|
|
],
|
|
tags = ["no_pip"],
|
|
)
|
|
|
|
filegroup(
|
|
name = "all_files",
|
|
srcs = glob(
|
|
["**/*"],
|
|
exclude = [
|
|
"**/METADATA",
|
|
"**/OWNERS",
|
|
],
|
|
),
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
)
|