polygraphy.backend.tf.loader.ModifyGraph Class Reference

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

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

Public Member Functions
def	__init__ (self, graph, outputs=None)
def	__call__ (self)

Public Attributes
	outputs

Private Attributes
	_graph

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.tf.loader.ModifyGraph.__init__	(		self,
			graph,
			outputs = None
	)

Functor that modifies 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.


    outputs (List[str]):
    Names of output tensors. If provided, this will override the outputs
    determined by the loader.
    If a value of `constants.MARK_ALL` is used instead of a list, all tensors in the network are marked.

Member Function Documentation

__call__()

def polygraphy.backend.tf.loader.ModifyGraph.__call__

(

self

)

Modifies 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.ModifyGraph._graph

private

outputs

polygraphy.backend.tf.loader.ModifyGraph.outputs

---

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

• backend/tf/loader.py