chore: import upstream snapshot with attribution
cffconvert / validate (push) Has been skipped
License Check / license-check (push) Failing after 2s

This commit is contained in:
wehub-resource-sync
2026-07-13 12:14:16 +08:00
commit 8a852e4b4e
36502 changed files with 9277225 additions and 0 deletions
@@ -0,0 +1,68 @@
# 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",
],
)