1# Copyright 2021 The TensorFlow Authors. All Rights Reserved. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================== 15 16# Description: 17# Keras saving and loading files for SavedModel. 18 19# buildifier: disable=same-origin-load 20 21package( 22 default_visibility = [ 23 "//tensorflow/python/keras/saving:__subpackages__", 24 ], 25 licenses = ["notice"], 26) 27 28py_library( 29 name = "load_context", 30 srcs = ["load_context.py"], 31 visibility = ["//tensorflow/python/distribute:__pkg__"], 32) 33 34py_library( 35 name = "saved_model", 36 srcs = [ 37 "base_serialization.py", 38 "constants.py", 39 "json_utils.py", 40 "layer_serialization.py", 41 "load.py", 42 "load_context.py", 43 "metric_serialization.py", 44 "model_serialization.py", 45 "network_serialization.py", 46 "save.py", 47 "save_impl.py", 48 "serialized_attributes.py", 49 "utils.py", 50 ], 51 srcs_version = "PY3", 52 deps = [ 53 "//tensorflow/python:constant_op", 54 "//tensorflow/python:dtypes", 55 "//tensorflow/python:framework_ops", 56 "//tensorflow/python:platform", 57 "//tensorflow/python:tensor_util", 58 "//tensorflow/python:util", 59 "//tensorflow/python/saved_model:signature_constants", 60 "//tensorflow/python/saved_model:signature_def_utils", 61 "//tensorflow/python/saved_model:tag_constants", 62 "//tensorflow/python/saved_model:utils", 63 ], 64) 65