Struct NetronExporter 

pub struct NetronExporter { /* private fields */ }

Netron exporter for model visualization

Implementations

impl NetronExporter

pub fn new(model_name: &str, description: &str) -> Self

Create a new Netron exporter

Arguments

• model_name - Name of the model
• description - Model description

Example

use trustformers_debug::NetronExporter;

let exporter = NetronExporter::new("bert-base", "BERT base model");

pub fn with_format(self, format: ExportFormat) -> Self

Set the export format

pub fn set_metadata(&mut self, metadata: ModelMetadata)

Set model metadata

pub fn set_author(&mut self, author: &str)

Add model author

pub fn set_version(&mut self, version: &str)

Add model version

pub fn add_property(&mut self, key: &str, value: &str)

Add a custom property to metadata

pub fn add_input(&mut self, name: &str, dtype: &str, shape: Vec<i64>)

Add an input tensor

pub fn add_output(&mut self, name: &str, dtype: &str, shape: Vec<i64>)

Add an output tensor

pub fn add_node( &mut self, name: &str, op_type: &str, inputs: Vec<String>, outputs: Vec<String>, attributes: HashMap<String, AttributeValue>, )

Add a graph node (layer/operation)

Example

let mut exporter = NetronExporter::new("model", "test model");

let mut attrs = HashMap::new();
attrs.insert("in_features".to_string(),
             trustformers_debug::netron_export::AttributeValue::Int(768));
attrs.insert("out_features".to_string(),
             trustformers_debug::netron_export::AttributeValue::Int(3072));

exporter.add_node(
    "fc1",
    "Linear",
    vec!["input".to_string()],
    vec!["hidden".to_string()],
    attrs,
);

pub fn add_node_with_doc( &mut self, name: &str, op_type: &str, inputs: Vec<String>, outputs: Vec<String>, attributes: HashMap<String, AttributeValue>, doc_string: &str, )

Add a node with documentation

pub fn add_tensor_data( &mut self, name: &str, dtype: &str, shape: Vec<i64>, data: Option<Vec<f32>>, )

Add tensor data (weights/biases)

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

Export the model to a file

Arguments

• path - Output file path

Example

exporter.export("model.json").unwrap();

pub fn model(&self) -> &NetronModel

Get a reference to the model

pub fn model_mut(&mut self) -> &mut NetronModel

Get a mutable reference to the model

pub fn to_json_string(&self) -> Result<String>

Export model to a string (JSON format)

pub fn create_linear_node( name: &str, input_name: &str, output_name: &str, in_features: i64, out_features: i64, has_bias: bool, ) -> GraphNode

Create a simple linear layer node

pub fn create_attention_node( name: &str, input_name: &str, output_name: &str, num_heads: i64, head_dim: i64, ) -> GraphNode

Create a transformer attention node

pub fn create_layernorm_node( name: &str, input_name: &str, output_name: &str, normalized_shape: Vec<i64>, eps: f64, ) -> GraphNode

Create a layer normalization node