tensorflow::ops::Restore

#include <io_ops.h>

Restores a tensor from checkpoint files.

Summary

Reads a tensor stored in one or several files. If there are several files (for instance because a tensor was saved as slices), file_pattern may contain wildcard symbols (* and ?) in the filename portion only, not in the directory portion.

If a file_pattern matches several files, preferred_shard can be used to hint in which file the requested tensor is likely to be found. This op will first open the file at index preferred_shard in the list of matching files and try to restore tensors from that file. Only if some tensors or tensor slices are not found in that first file, then the Op opens all the files. Setting preferred_shard to match the value passed as the shard input of a matching Save Op may speed up Restore. This attribute only affects performance, not correctness. The default value -1 means files are processed in order.

See also RestoreSlice.

Arguments:

  • scope: A Scope object
  • file_pattern: Must have a single element. The pattern of the files from which we read the tensor.
  • tensor_name: Must have a single element. The name of the tensor to be restored.
  • dt: The type of the tensor to be restored.

Optional attributes (see Attrs):

  • preferred_shard: Index of file to open first if multiple files match file_pattern.

Returns:

  • Output: The restored tensor.
Constructors and Destructors
Restore(const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt)
Restore(const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs)
Public attributes
operation
tensor
Public functions
node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const
Public static functions
PreferredShard(int64 x)
Structs
tensorflow::ops::Restore::Attrs

Optional attribute setters for Restore.

Public attributes

operation

Operation operation

tensor

::tensorflow::Output tensor

Public functions

Restore

 Restore(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input file_pattern,
  ::tensorflow::Input tensor_name,
  DataType dt
)

Restore

 Restore(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input file_pattern,
  ::tensorflow::Input tensor_name,
  DataType dt,
  const Restore::Attrs & attrs
)

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

operator::tensorflow::Input() const 

operator::tensorflow::Output

operator::tensorflow::Output() const 

Public static functions

PreferredShard

Attrs PreferredShard(
  int64 x
)

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/cc/class/tensorflow/ops/restore