WeHub Mirror
16d3f15320
ARM CI / build (3.10) (push) Has been cancelled
cffconvert / validate (push) Has been cancelled
Creates a GitHub Issue when a PR Rolled back via Commit to Master / create-issue-on-pr-rollback (push) Has been cancelled
Scorecards supply-chain security / Scorecards analysis (push) Has been cancelled
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
# Copyright 2026 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
|
|
# Description:
|
|
# Contains the Keras save model API (internal TensorFlow version).
|
|
|
|
load("@xla//third_party/rules_python/python:defs.bzl", "py_library")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
|
|
# TODO(scottzhu): Remove non-keras deps from TF.
|
|
default_visibility = [
|
|
"//tensorflow/python/distribute:__pkg__",
|
|
"//tensorflow/python/keras:__subpackages__",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
filegroup(
|
|
name = "all_py_srcs",
|
|
srcs = glob([
|
|
"*.py",
|
|
"**/*.py",
|
|
]),
|
|
visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
|
|
)
|
|
|
|
py_library(
|
|
name = "saving",
|
|
srcs = [
|
|
"__init__.py",
|
|
"hdf5_format.py",
|
|
"model_config.py",
|
|
"save.py",
|
|
"saved_model_experimental.py",
|
|
"saving_utils.py",
|
|
],
|
|
srcs_version = "PY3",
|
|
strict_deps = False,
|
|
deps = [
|
|
"//tensorflow/python/checkpoint:graph_view",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/keras:backend",
|
|
"//tensorflow/python/keras:losses",
|
|
"//tensorflow/python/keras:optimizers",
|
|
"//tensorflow/python/keras:regularizers",
|
|
"//tensorflow/python/keras/engine:input_spec",
|
|
"//tensorflow/python/keras/mixed_precision:autocast_variable",
|
|
"//tensorflow/python/keras/protobuf:saved_metadata_proto_py",
|
|
"//tensorflow/python/keras/saving/saved_model",
|
|
"//tensorflow/python/keras/saving/utils_v1",
|
|
"//tensorflow/python/keras/utils:engine_utils",
|
|
"//tensorflow/python/keras/utils:metrics_utils",
|
|
"//tensorflow/python/keras/utils:mode_keys",
|
|
"//tensorflow/python/lib/io:file_io",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/platform:gfile",
|
|
"//tensorflow/python/platform:tf_logging",
|
|
"//tensorflow/python/saved_model",
|
|
"//tensorflow/python/training:saver",
|
|
"@pypi//h5py",
|
|
],
|
|
)
|