Update ops-related pbtxt files.
Change: 134133051
This commit is contained in:
parent
86c6c4165b
commit
bd2a141574
tensorflow/core/ops
@ -16182,6 +16182,24 @@ op {
|
||||
minimum: 1
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MergeV2Checkpoints"
|
||||
input_arg {
|
||||
name: "checkpoint_prefixes"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "destination_prefix"
|
||||
type: DT_STRING
|
||||
}
|
||||
attr {
|
||||
name: "delete_old_dirs"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "Min"
|
||||
input_arg {
|
||||
@ -21225,6 +21243,31 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "RestoreV2"
|
||||
input_arg {
|
||||
name: "prefix"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "tensor_names"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "shape_and_slices"
|
||||
type: DT_STRING
|
||||
}
|
||||
output_arg {
|
||||
name: "tensors"
|
||||
type_list_attr: "dtypes"
|
||||
}
|
||||
attr {
|
||||
name: "dtypes"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "Reverse"
|
||||
input_arg {
|
||||
@ -21818,6 +21861,31 @@ op {
|
||||
minimum: 1
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "SaveV2"
|
||||
input_arg {
|
||||
name: "prefix"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "tensor_names"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "shape_and_slices"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "tensors"
|
||||
type_list_attr: "dtypes"
|
||||
}
|
||||
attr {
|
||||
name: "dtypes"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "ScalarSummary"
|
||||
input_arg {
|
||||
|
@ -9319,6 +9319,29 @@ op {
|
||||
summary: "Merges summaries."
|
||||
description: "This op creates a\n[`Summary`](https://www.tensorflow.org/code/tensorflow/core/framework/summary.proto)\nprotocol buffer that contains the union of all the values in the input\nsummaries.\n\nWhen the Op is run, it reports an `InvalidArgument` error if multiple values\nin the summaries to merge use the same tag."
|
||||
}
|
||||
op {
|
||||
name: "MergeV2Checkpoints"
|
||||
input_arg {
|
||||
name: "checkpoint_prefixes"
|
||||
description: "prefixes of V2 checkpoints to merge."
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "destination_prefix"
|
||||
description: "scalar. The desired final prefix. Allowed to be the same\nas one of the checkpoint_prefixes."
|
||||
type: DT_STRING
|
||||
}
|
||||
attr {
|
||||
name: "delete_old_dirs"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
description: "see above."
|
||||
}
|
||||
summary: "V2 format specific: merges the metadata files of sharded checkpoints. The"
|
||||
description: "result is one logical checkpoint, with one physical metadata file and renamed\ndata files.\n\nIntended for \"grouping\" multiple checkpoints in a sharded checkpoint setup.\n\nIf delete_old_dirs is true, attempts to delete recursively the dirname of each\npath in the input checkpoint_prefixes. This is useful when those paths are non\nuser-facing temporary locations."
|
||||
}
|
||||
op {
|
||||
name: "Min"
|
||||
input_arg {
|
||||
@ -12438,6 +12461,38 @@ op {
|
||||
summary: "Restores a tensor from checkpoint files."
|
||||
description: "This is like `Restore` except that restored tensor can be listed as filling\nonly a slice of a larger tensor. `shape_and_slice` specifies the shape of the\nlarger tensor and the slice that the restored tensor covers.\n\nThe `shape_and_slice` input has the same format as the\nelements of the `shapes_and_slices` input of the `SaveSlices` op."
|
||||
}
|
||||
op {
|
||||
name: "RestoreV2"
|
||||
input_arg {
|
||||
name: "prefix"
|
||||
description: "scalar. The prefix of a V2 checkpoint."
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "tensor_names"
|
||||
description: "shape {N}. The names of the tensors to be restored."
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "shape_and_slices"
|
||||
description: "shape {N}. The slice specs of the tensors to be restored.\nEmpty strings indicate that they are non-partitioned tensors."
|
||||
type: DT_STRING
|
||||
}
|
||||
output_arg {
|
||||
name: "tensors"
|
||||
description: "shape {N}. The restored tensors, whose shapes are read from the\ncheckpoint directly."
|
||||
type_list_attr: "dtypes"
|
||||
}
|
||||
attr {
|
||||
name: "dtypes"
|
||||
type: "list(type)"
|
||||
description: "shape {N}. The list of expected dtype for the tensors. Must match\nthose stored in the checkpoint."
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
summary: "Restores tensors from a V2 checkpoint."
|
||||
description: "For backward compatibility with the V1 format, this Op currently allows\nrestoring from a V1 checkpoint as well:\n - This Op first attempts to find the V2 index file pointed to by \"prefix\", and\n if found proceed to read it as a V2 checkpoint;\n - Otherwise the V1 read path is invoked.\nRelying on this behavior is not recommended, as the ability to fall back to read\nV1 might be deprecated and eventually removed.\n\nBy default, restores the named tensors in full. If the caller wishes to restore\nspecific slices of stored tensors, \"shape_and_slices\" should be non-empty\nstrings and correspondingly well-formed.\n\nCallers must ensure all the named tensors are indeed stored in the checkpoint."
|
||||
}
|
||||
op {
|
||||
name: "Reverse"
|
||||
input_arg {
|
||||
@ -12745,6 +12800,37 @@ op {
|
||||
summary: "Saves input tensors slices to disk."
|
||||
description: "This is like `Save` except that tensors can be listed in the saved file as being\na slice of a larger tensor. `shapes_and_slices` specifies the shape of the\nlarger tensor and the slice that this tensor covers. `shapes_and_slices` must\nhave as many elements as `tensor_names`.\n\nElements of the `shapes_and_slices` input must either be:\n\n* The empty string, in which case the corresponding tensor is\n saved normally.\n* A string of the form `dim0 dim1 ... dimN-1 slice-spec` where the\n `dimI` are the dimensions of the larger tensor and `slice-spec`\n specifies what part is covered by the tensor to save.\n\n`slice-spec` itself is a `:`-separated list: `slice0:slice1:...:sliceN-1`\nwhere each `sliceI` is either:\n\n* The string `-` meaning that the slice covers all indices of this dimension\n* `start,length` where `start` and `length` are integers. In that\n case the slice covers `length` indices starting at `start`.\n\nSee also `Save`."
|
||||
}
|
||||
op {
|
||||
name: "SaveV2"
|
||||
input_arg {
|
||||
name: "prefix"
|
||||
description: "scalar. The prefix of the V2 checkpoint to which we write the tensors."
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "tensor_names"
|
||||
description: "shape {N}. The names of the tensors to be saved."
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "shape_and_slices"
|
||||
description: "shape {N}. The slice specs of the tensors to be saved.\nEmpty strings indicate that they are non-partitioned tensors."
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "tensors"
|
||||
description: "`N` tensors to save."
|
||||
type_list_attr: "dtypes"
|
||||
}
|
||||
attr {
|
||||
name: "dtypes"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
summary: "Saves tensors in V2 checkpoint format."
|
||||
description: "By default, saves the named tensors in full. If the caller wishes to save\nspecific slices of full tensors, \"shape_and_slices\" should be non-empty strings\nand correspondingly well-formed."
|
||||
}
|
||||
op {
|
||||
name: "ScalarSummary"
|
||||
input_arg {
|
||||
|
Loading…
Reference in New Issue
Block a user