class SambaGraph(name: str)#

New in version 1.18.

Base class for representing a portion of a model’s computational graph.


name – the name of the graph

class FwdGraph(start: Tuple[SambaTensor], stop: Tuple[SambaTensor], name: str)#

Bases: SambaGraph

New in version 1.18.

Specify all or part of the forward pass of a computational graph. Multiple FwdGraphs can be combined to fully represent the forward pass of the computational graph. Use trace_multigraph to trace the graph and create a handle to the graph’s output tensors, which is required for creating a FwdGraph object.

  • start – the input tensors of the graph

  • stop – the output tensors of the graph

  • name – the name of the graph