Struct GraphVisualizer 

pub struct GraphVisualizer { /* private fields */ }

Graph visualizer for computation graphs

Implementations

impl GraphVisualizer

pub fn new(graph_name: &str) -> Self

Create a new graph visualizer

Arguments

• graph_name - Name of the computation graph

Example

use trustformers_debug::GraphVisualizer;

let visualizer = GraphVisualizer::new("my_model");

pub fn with_config(graph_name: &str, config: GraphVisualizerConfig) -> Self

Create a graph visualizer with custom configuration

pub fn add_node( &mut self, id: &str, label: &str, op_type: &str, shape: Option<Vec<i64>>, dtype: Option<String>, attributes: HashMap<String, String>, )

Add a node to the graph

Example

visualizer.add_node(
    "layer1",
    "Linear Layer 1",
    "Linear",
    Some(vec![10, 20]),
    Some("float32".to_string()),
    HashMap::new(),
);

pub fn add_edge( &mut self, from: &str, to: &str, label: Option<String>, shape: Option<Vec<i64>>, )

Add an edge to the graph

pub fn mark_input(&mut self, node_id: &str)

Mark a node as an input

pub fn mark_output(&mut self, node_id: &str)

Mark a node as an output

pub fn export_to_dot<P: AsRef<Path>>(&mut self, path: P) -> Result<()>

Export graph to GraphViz DOT format

Example

visualizer.export_to_dot("model.dot").unwrap();

pub fn export_to_json<P: AsRef<Path>>(&self, path: P) -> Result<()>

Export graph to JSON format

pub fn statistics(&self) -> GraphStatistics

Get statistics about the graph

pub fn summary(&self) -> String

Print a summary of the graph

Trait Implementations

impl Debug for GraphVisualizer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.