View source on GitHub
|
Options for loading a SavedModel.
tf.saved_model.LoadOptions(
experimental_io_device=None
)
This function may be used in the options argument in functions that
load a SavedModel (tf.saved_model.load, tf.keras.models.load_model).
Args | |
|---|---|
experimental_io_device
|
string. Applies in a distributed setting.
Tensorflow device to use to access the filesystem. If None (default)
then for each variable the filesystem is accessed from the CPU:0 device
of the host where that variable is assigned. If specified, the
filesystem is instead accessed from that device for all variables.
This is for example useful if you want to load from a local directory,
such as "/tmp" when running in a distributed setting. In that case
pass a device for the host where the "/tmp" directory is accessible.
|