The metadata of each Keras model/layer will be moved from the SavedModel to a separate keras.metadata file. PiperOrigin-RevId: 337394096 Change-Id: Iccba44c8478e038fc9de036313c2e7e88fd5605d
24 lines
544 B
Python
24 lines
544 B
Python
# Description:
|
|
# Contains Keras protobufs
|
|
load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library")
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//tensorflow/python/keras:__subpackages__",
|
|
],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
tf_proto_library(
|
|
name = "projector_config_proto",
|
|
srcs = ["projector_config.proto"],
|
|
cc_api_version = 2,
|
|
)
|
|
|
|
tf_proto_library(
|
|
name = "saved_metadata_proto",
|
|
srcs = ["saved_metadata.proto"],
|
|
cc_api_version = 2,
|
|
protodeps = ["//tensorflow/core:protos_all"],
|
|
)
|