analogvnn.graph.ModelGraph#

Module Contents#

Classes#

ModelGraph

Store model's graph.

class analogvnn.graph.ModelGraph.ModelGraph(use_autograd_graph: bool = False, allow_loops: bool = False)[source]#

Bases: analogvnn.graph.ModelGraphState.ModelGraphState

Store model’s graph.

Variables:
  • forward_graph (ForwardGraph) – store model’s forward graph.

  • backward_graph (BackwardGraph) – store model’s backward graph.

forward_graph: analogvnn.graph.ForwardGraph.ForwardGraph[source]#
backward_graph: analogvnn.graph.BackwardGraph.BackwardGraph[source]#
compile(is_static: bool = True, auto_backward_graph: bool = False) ModelGraph[source]#

Compile the model graph.

Parameters:
  • is_static (bool) – If True, the model graph is static.

  • auto_backward_graph (bool) – If True, the backward graph is automatically created.

Returns:

self.

Return type:

ModelGraph