polygraphy.backend.tf.loader.SaveGraph Class Reference

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

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

Public Member Functions
def	__init__ (self, graph, path=None, tensorboard_dir=None, engine_dir=None)
def	__call__ (self)

Public Attributes
	path
	tensorboard_dir
	engine_dir

Private Attributes
	_graph

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.tf.loader.SaveGraph.__init__	(		self,
			graph,
			path = None,
			tensorboard_dir = None,
			engine_dir = None
	)

Functor that writes out artifacts from a TensorFlow graph.

Args:
    graph (Callable() -> Tuple[tf.Graph, Sequence[str]]):
    A callable that can supply a tuple containing a
    TensorFlow graph and output names.


    path (str): Path at which to save the frozen graphdef.
    tensorboard_dir (str): The directory in which to write TensorBoard visualizations.
    engine_dir (str): The directory in which to save TF-TRT engines,

Member Function Documentation

__call__()

def polygraphy.backend.tf.loader.SaveGraph.__call__

(

self

)

Writes out artifacts from a TensorFlow Graph.

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

Member Data Documentation

_graph

polygraphy.backend.tf.loader.SaveGraph._graph

private

path

polygraphy.backend.tf.loader.SaveGraph.path

tensorboard_dir

polygraphy.backend.tf.loader.SaveGraph.tensorboard_dir

engine_dir

polygraphy.backend.tf.loader.SaveGraph.engine_dir

---

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

• backend/tf/loader.py