analogvnn.graph.ModelGraph
#
Module Contents#
Classes#
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:
- Returns:
self.
- Return type: