Skip to main content

graph_to_pytorch_code

Function graph_to_pytorch_code 

Source
pub fn graph_to_pytorch_code(
    graph: &FxGraph,
    model_name: &str,
) -> Result<String>
Expand description

Convert FxGraph to PyTorch model code