ee1dcbbd66
Adds a slot `function_aliases` to tf.saved_model.SaveOption to allow users to provide aliases for FunctionDef in SavedModel. This will allow them to specify FunctionDef with their customized alias in the use of SavedModel. E.g. When trying to rewrite a FunctionDef for TPU inference they can directly use alias instead of the name of FunctionDef, which could be something like "__inference_serve_<lambda>_1234". This piece of information will be recorded in a new field `function_aliases` in MetaGraphDef.meta_info_def, which will later be consumed by external tools. PiperOrigin-RevId: 289542552 Change-Id: I3690532034c3d3a644ce9f79777c7cdb0bf563bd
76 lines
1.5 KiB
Plaintext
76 lines
1.5 KiB
Plaintext
path: "tensorflow.MetaGraphDef.MetaInfoDef"
|
|
tf_proto {
|
|
descriptor {
|
|
name: "MetaInfoDef"
|
|
field {
|
|
name: "meta_graph_version"
|
|
number: 1
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_STRING
|
|
}
|
|
field {
|
|
name: "stripped_op_list"
|
|
number: 2
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_MESSAGE
|
|
type_name: ".tensorflow.OpList"
|
|
}
|
|
field {
|
|
name: "any_info"
|
|
number: 3
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_MESSAGE
|
|
type_name: ".google.protobuf.Any"
|
|
}
|
|
field {
|
|
name: "tags"
|
|
number: 4
|
|
label: LABEL_REPEATED
|
|
type: TYPE_STRING
|
|
}
|
|
field {
|
|
name: "tensorflow_version"
|
|
number: 5
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_STRING
|
|
}
|
|
field {
|
|
name: "tensorflow_git_version"
|
|
number: 6
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_STRING
|
|
}
|
|
field {
|
|
name: "stripped_default_attrs"
|
|
number: 7
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_BOOL
|
|
}
|
|
field {
|
|
name: "function_aliases"
|
|
number: 8
|
|
label: LABEL_REPEATED
|
|
type: TYPE_MESSAGE
|
|
type_name: ".tensorflow.MetaGraphDef.MetaInfoDef.FunctionAliasesEntry"
|
|
}
|
|
nested_type {
|
|
name: "FunctionAliasesEntry"
|
|
field {
|
|
name: "key"
|
|
number: 1
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_STRING
|
|
}
|
|
field {
|
|
name: "value"
|
|
number: 2
|
|
label: LABEL_OPTIONAL
|
|
type: TYPE_STRING
|
|
}
|
|
options {
|
|
map_entry: true
|
|
}
|
|
}
|
|
}
|
|
}
|