wehub-resource-sync
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
# Description:
|
|
# Python library for splitting and joining large protos.
|
|
load("//tensorflow:pytype.default.bzl", "pytype_strict_library")
|
|
load("//tensorflow:tensorflow.bzl", "py_test")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
|
|
default_visibility = ["//tensorflow:internal"],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
pytype_strict_library(
|
|
name = "saved_model",
|
|
srcs = ["saved_model.py"],
|
|
# NOTE(yibaimeng): To be removed when everything is migrated to `pywrap_saved_model.Save`.
|
|
visibility = [
|
|
"//tensorflow:internal",
|
|
"//waymo/ml/deploy/tensorflow:__pkg__",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/tools/proto_splitter:constants",
|
|
"//tensorflow/tools/proto_splitter:split",
|
|
"//tensorflow/tools/proto_splitter:split_graph_def",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "saved_model_test",
|
|
srcs = ["saved_model_test.py"],
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_mac", # b/291933687
|
|
"no_windows", # b/291001524
|
|
],
|
|
deps = [
|
|
":saved_model",
|
|
":test_util",
|
|
#internal proto upb dep
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python/platform:client_testlib",
|
|
"//tensorflow/tools/proto_splitter:constants",
|
|
],
|
|
)
|
|
|
|
pytype_strict_library(
|
|
name = "test_util",
|
|
srcs = ["test_util.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "test_util_test",
|
|
srcs = ["test_util_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
#internal proto upb dep
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/platform:client_testlib",
|
|
"//tensorflow/tools/proto_splitter/python:test_util",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|