polygraphy.backend.tf.loader.GraphFromCkpt Class Reference

Inheritance diagram for polygraphy.backend.tf.loader.GraphFromCkpt:

Collaboration diagram for polygraphy.backend.tf.loader.GraphFromCkpt:

Public Member Functions
def	__init__ (self, dir, name=None)
def	__call__ (self)

Public Attributes
	dir
	name

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.tf.loader.GraphFromCkpt.__init__	(		self,
			dir,
			name = None
	)

Functor that loads a TensorFlow model from a checkpoint.  Note that in order to use checkpoints,
you must NOT use subprocesses in the Comparator.

Args:
    dir (str): Path to a directory containing checkpoints.


    name (str):
    The name of the checkpoint to load, not including the file extension.
    For example, to load `model.meta`, the argument would be `model`.

Member Function Documentation

__call__()

def polygraphy.backend.tf.loader.GraphFromCkpt.__call__

(

self

)

Loads a TensorFlow model from a checkpoint.

Returns:
    Tuple[tf.Graph, Sequence[str]]: The TensorFlow graph, and the names of its outputs.

Member Data Documentation

dir

polygraphy.backend.tf.loader.GraphFromCkpt.dir

name

polygraphy.backend.tf.loader.GraphFromCkpt.name

---

The documentation for this class was generated from the following file:

• backend/tf/loader.py