wehub-resource-sync
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
# 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",
|
|
],
|
|
)
|