Update ops-related pbtxt files.
PiperOrigin-RevId: 166962648
This commit is contained in:
parent
c82d70d383
commit
9c0302760d
@ -22394,6 +22394,18 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "RestoreIterator"
|
||||
input_arg {
|
||||
name: "iterator"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "path"
|
||||
type: DT_STRING
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "RestoreSlice"
|
||||
input_arg {
|
||||
@ -23117,6 +23129,18 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "SaveIterator"
|
||||
input_arg {
|
||||
name: "iterator"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "path"
|
||||
type: DT_STRING
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "SaveSlices"
|
||||
input_arg {
|
||||
|
@ -22093,6 +22093,19 @@ op {
|
||||
description: "Reads a tensor stored in one or several files. If there are several files (for\ninstance because a tensor was saved as slices), `file_pattern` may contain\nwildcard symbols (`*` and `?`) in the filename portion only, not in the\ndirectory portion.\n\nIf a `file_pattern` matches several files, `preferred_shard` can be used to hint\nin which file the requested tensor is likely to be found. This op will first\nopen the file at index `preferred_shard` in the list of matching files and try\nto restore tensors from that file. Only if some tensors or tensor slices are\nnot found in that first file, then the Op opens all the files. Setting\n`preferred_shard` to match the value passed as the `shard` input\nof a matching `Save` Op may speed up Restore. This attribute only affects\nperformance, not correctness. The default value -1 means files are processed in\norder.\n\nSee also `RestoreSlice`."
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "RestoreIterator"
|
||||
input_arg {
|
||||
name: "iterator"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "path"
|
||||
type: DT_STRING
|
||||
}
|
||||
summary: "Restores the state of the `iterator` from the checkpoint saved at `path` using \"SaveIterator\"."
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "RestoreSlice"
|
||||
input_arg {
|
||||
@ -22653,6 +22666,20 @@ op {
|
||||
description: "The size of `tensor_names` must match the number of tensors in `data`. `data[i]`\nis written to `filename` with name `tensor_names[i]`.\n\nSee also `SaveSlices`."
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "SaveIterator"
|
||||
input_arg {
|
||||
name: "iterator"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "path"
|
||||
type: DT_STRING
|
||||
}
|
||||
summary: "Saves the state of the `iterator` at `path`."
|
||||
description: "This state can be restored using \"RestoreIterator\"."
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "SaveSlices"
|
||||
input_arg {
|
||||
|
Loading…
x
Reference in New Issue
Block a user