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
+291
View File
@@ -0,0 +1,291 @@
# Description:
# Trackable class and subclass definitions.
load("@xla//third_party/rules_python/python:py_library.bzl", "py_library")
load("//tensorflow:tensorflow.default.bzl", "tf_py_strict_test")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = [
"//tensorflow:internal",
],
licenses = ["notice"],
)
py_library(
name = "trackable",
strict_deps = True,
visibility = [
"//tensorflow:internal",
"//third_party/py/tf_agents:__subpackages__",
],
deps = [
":asset",
":autotrackable",
":base",
":base_delegate",
":constants",
":converter",
":data_structures",
":layer_utils",
":python_state",
":resource",
":trackable_init",
":trackable_utils",
],
)
py_library(
name = "trackable_init",
srcs = ["__init__.py"],
strict_deps = True,
)
py_library(
name = "base",
srcs = ["base.py"],
strict_deps = True,
deps = [
":constants",
"//tensorflow/python/eager:context",
"//tensorflow/python/framework:dtypes",
"//tensorflow/python/framework:ops",
"//tensorflow/python/ops:control_flow_ops_gen",
"//tensorflow/python/training/saving:saveable_object",
"//tensorflow/python/util:tf_contextlib",
"//tensorflow/python/util:tf_decorator_py",
"//tensorflow/python/util:tf_export",
],
)
tf_py_strict_test(
name = "base_test",
srcs = ["base_test.py"],
deps = [
":base",
"//tensorflow/python/checkpoint",
"//tensorflow/python/framework:ops",
"//tensorflow/python/ops:variable_scope",
"//tensorflow/python/platform:client_testlib",
],
)
py_library(
name = "constants",
srcs = ["constants.py"],
strict_deps = True,
)
py_library(
name = "converter",
srcs = ["converter.py"],
strict_deps = True,
deps = [
":base",
":data_structures",
"//tensorflow/python/eager/polymorphic_function:saved_model_utils",
"//tensorflow/python/framework:dtypes",
"//tensorflow/python/framework:tensor_util",
"//tensorflow/python/ops:resource_variable_ops",
],
)
py_library(
name = "trackable_utils",
srcs = ["trackable_utils.py"],
strict_deps = True,
)
tf_py_strict_test(
name = "trackable_utils_test",
srcs = ["trackable_utils_test.py"],
deps = [
":trackable_utils",
"//tensorflow/python/eager:test",
],
)
py_library(
name = "base_delegate",
srcs = ["base_delegate.py"],
strict_deps = True,
deps = [
"//tensorflow/python/util:tf_export",
],
)
tf_py_strict_test(
name = "base_delegate_test",
srcs = ["base_delegate_test.py"],
deps = [
":base",
":base_delegate",
"//tensorflow/python:extra_py_tests_deps",
"//tensorflow/python/checkpoint",
"//tensorflow/python/checkpoint:checkpoint_options",
"//tensorflow/python/eager:test",
"//tensorflow/python/framework:test_lib",
"//tensorflow/python/ops:variables",
"//tensorflow/python/saved_model:load",
"//tensorflow/python/saved_model:save",
"@absl_py//absl/testing:parameterized",
],
)
py_library(
name = "asset",
srcs = ["asset.py"],
strict_deps = True,
deps = [
":base",
"//tensorflow/python/eager:context",
"//tensorflow/python/framework:dtypes",
"//tensorflow/python/framework:ops",
"//tensorflow/python/framework:tensor_conversion_registry",
"//tensorflow/python/lib/io:file_io",
"//tensorflow/python/ops:array_ops",
"//tensorflow/python/ops:resource_variable_ops",
"//tensorflow/python/saved_model:path_helpers",
"//tensorflow/python/util:tf_export",
],
)
py_library(
name = "autotrackable",
srcs = ["autotrackable.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":base",
":data_structures",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:function",
"//tensorflow/python/types:core",
"//tensorflow/python/util:tf_export",
"@absl_py//absl/logging",
],
)
tf_py_strict_test(
name = "autotrackable_test",
srcs = ["autotrackable_test.py"],
deps = [
":autotrackable",
":data_structures",
"//tensorflow/python/checkpoint",
"//tensorflow/python/framework:test_lib",
"//tensorflow/python/ops:array_ops",
"//tensorflow/python/platform:client_testlib",
"//tensorflow/python/util:nest",
],
)
py_library(
name = "resource",
srcs = ["resource.py"],
strict_deps = True,
visibility = [
"//intelligence/climate_foundations/vlm/export:__subpackages__",
"//tensorflow:internal",
],
deps = [
":base",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/framework:ops",
"//tensorflow/python/framework:tensor",
"//tensorflow/python/util:tf_contextlib",
"//tensorflow/python/util:tf_export",
],
)
tf_py_strict_test(
name = "resource_test",
srcs = ["resource_test.py"],
deps = [
":resource",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:wrap_function",
"//tensorflow/python/framework:ops",
"//tensorflow/python/framework:test_lib",
"//tensorflow/python/platform:client_testlib",
],
)
py_library(
name = "layer_utils",
srcs = ["layer_utils.py"],
strict_deps = True,
deps = ["//tensorflow/python/util:object_identity"],
)
py_library(
name = "data_structures",
srcs = ["data_structures.py"],
strict_deps = True,
deps = [
":base",
":layer_utils",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:function",
"//tensorflow/python/ops:variables",
"//tensorflow/python/util:compat",
"//tensorflow/python/util:tf_export",
"@pypi//wrapt",
],
)
tf_py_strict_test(
name = "data_structures_test",
srcs = ["data_structures_test.py"],
tags = [
"no_oss", # Keras is not available in OSS test
"no_windows",
"nomac",
],
deps = [
":autotrackable",
":data_structures",
"//tensorflow/python/checkpoint",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test",
"//tensorflow/python/framework:constant_op",
"//tensorflow/python/framework:tensor_shape",
"//tensorflow/python/module",
"//tensorflow/python/ops:array_ops",
"//tensorflow/python/ops:resource_variable_ops",
"//tensorflow/python/ops:variables",
"//tensorflow/python/util:nest",
"//tensorflow/python/util:serialization",
"@absl_py//absl/testing:parameterized",
],
)
py_library(
name = "python_state",
srcs = ["python_state.py"],
strict_deps = True,
deps = [
":base",
"//tensorflow/python/framework:constant_op",
"//tensorflow/python/framework:dtypes",
"//tensorflow/python/framework:ops",
"//tensorflow/python/util:tf_export",
],
)
tf_py_strict_test(
name = "python_state_test",
srcs = ["python_state_test.py"],
deps = [
":python_state",
"//tensorflow/python/checkpoint",
"//tensorflow/python/client:session",
"//tensorflow/python/framework:ops",
"//tensorflow/python/framework:test_lib",
"//tensorflow/python/module",
"//tensorflow/python/platform:client_testlib",
],
)