Skip to main content

ScriptableModule

torsh_jit

Trait ScriptableModule 

Source 
pub trait ScriptableModule {
    // Required method
    fn to_graph(&self) -> JitResult<ComputationGraph>;
}
Expand description

Trait for scriptable modules

Required Methods§

Source

fn to_graph(&self) -> JitResult<ComputationGraph>

Get the computation graph for this module

Implementors§