package(default_visibility = [
|
"//visibility:public",
|
])
|
|
licenses(["notice"]) # Apache 2.0
|
|
load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite")
|
load("//tensorflow:tensorflow.bzl", "tf_cc_binary")
|
|
common_copts = ["-Wall"]
|
|
py_binary(
|
name = "visualize",
|
srcs = ["visualize.py"],
|
data = [
|
"//tensorflow/lite/schema:schema.fbs",
|
"//tensorflow/python:platform",
|
"@flatbuffers//:flatc",
|
],
|
srcs_version = "PY2AND3",
|
)
|
|
tf_cc_binary(
|
name = "generate_op_registrations",
|
srcs = ["gen_op_registration_main.cc"],
|
deps = [
|
"//tensorflow/core:framework_internal",
|
"//tensorflow/core:lib",
|
"//tensorflow/lite/tools:gen_op_registration",
|
"@com_google_absl//absl/strings",
|
],
|
)
|
|
cc_library(
|
name = "gen_op_registration",
|
srcs = ["gen_op_registration.cc"],
|
hdrs = ["gen_op_registration.h"],
|
deps = [
|
"//tensorflow/lite:framework",
|
"//tensorflow/lite:string",
|
"@com_googlesource_code_re2//:re2",
|
],
|
)
|
|
cc_test(
|
name = "gen_op_registration_test",
|
srcs = ["gen_op_registration_test.cc"],
|
data = [
|
"//tensorflow/lite:testdata/0_subgraphs.bin",
|
"//tensorflow/lite:testdata/2_subgraphs.bin",
|
"//tensorflow/lite:testdata/empty_model.bin",
|
"//tensorflow/lite:testdata/test_model.bin",
|
"//tensorflow/lite:testdata/test_model_broken.bin",
|
],
|
tags = [
|
"tflite_not_portable_android",
|
"tflite_not_portable_ios",
|
],
|
deps = [
|
":gen_op_registration",
|
"@com_google_googletest//:gtest",
|
],
|
)
|
|
cc_library(
|
name = "verifier",
|
srcs = ["verifier.cc"],
|
hdrs = ["verifier.h"],
|
deps = [
|
"//tensorflow/lite:framework",
|
"//tensorflow/lite:schema_fbs_version",
|
"//tensorflow/lite:string_util",
|
"//tensorflow/lite/c:c_api_internal",
|
"//tensorflow/lite/schema:schema_fbs",
|
"@com_google_absl//absl/container:flat_hash_set",
|
],
|
)
|
|
cc_test(
|
name = "verifier_test",
|
size = "small",
|
srcs = ["verifier_test.cc"],
|
tags = [
|
"tflite_not_portable",
|
],
|
deps = [
|
":verifier",
|
"//tensorflow/core:framework_lite",
|
"//tensorflow/lite:framework",
|
"//tensorflow/lite:schema_fbs_version",
|
"//tensorflow/lite/c:c_api_internal",
|
"//tensorflow/lite/schema:schema_fbs",
|
"//tensorflow/lite/testing:util",
|
"@com_google_googletest//:gtest",
|
"@flatbuffers",
|
],
|
)
|
|
tflite_portable_test_suite()
|
