xref: /aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1# Description:
2#   Contains the Keras save model API (internal TensorFlow version).
3
4package(
5    # TODO(scottzhu): Remove non-keras deps from TF.
6    default_visibility = [
7        "//tensorflow/python/distribute:__pkg__",
8        "//tensorflow/python/keras:__subpackages__",
9    ],
10    licenses = ["notice"],
11)
12
13filegroup(
14    name = "all_py_srcs",
15    srcs = glob([
16        "*.py",
17        "**/*.py",
18    ]),
19    visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
20)
21
22py_library(
23    name = "saving",
24    srcs = [
25        "__init__.py",
26        "hdf5_format.py",
27        "model_config.py",
28        "save.py",
29        "saved_model_experimental.py",
30        "saving_utils.py",
31    ],
32    srcs_version = "PY3",
33    deps = [
34        "//tensorflow/python:lib",
35        "//tensorflow/python:math_ops",
36        "//tensorflow/python:platform",
37        "//tensorflow/python:saver",
38        "//tensorflow/python:tensor_spec",
39        "//tensorflow/python/checkpoint:graph_view",
40        "//tensorflow/python/eager:def_function",
41        "//tensorflow/python/keras:backend",
42        "//tensorflow/python/keras:losses",
43        "//tensorflow/python/keras:optimizers",
44        "//tensorflow/python/keras:regularizers",
45        "//tensorflow/python/keras/engine:input_spec",
46        "//tensorflow/python/keras/mixed_precision:autocast_variable",
47        "//tensorflow/python/keras/protobuf:saved_metadata_proto_py",
48        "//tensorflow/python/keras/saving/saved_model",
49        "//tensorflow/python/keras/saving/utils_v1",
50        "//tensorflow/python/keras/utils:engine_utils",
51        "//tensorflow/python/keras/utils:metrics_utils",
52        "//tensorflow/python/keras/utils:mode_keys",
53        "//tensorflow/python/saved_model",
54    ],
55)
56