Struct Graph

pub struct Graph<T: DType> { /* private fields */ }

Implementations

impl<T: DType> Graph<T>

pub fn empty() -> Self

Create an empty Graph

pub fn get_ops(&self) -> RwLockReadGuard<'_, Vec<GraphNode<T>>>

Read-only access to the list of operations

pub fn to_petgraph(&self) -> PetGraph<String, String>

pub fn to_dot(&self) -> String

Produce a DOT format string of this graph.

pub fn visualize<P: AsRef<Path>>(&self, filename: P) -> Result<()>

Visualize the graph by saving it to this file.

Install graphvis:

• brew install graphviz
• apt install graphviz

pub fn optimize(&mut self)

Optimize this graph.

Apply the following optimizations:

• Constant folding of elementwise fills
• Fuse mul-add into FMA
• Inplace binary operations when safe
• Inplace fused multiply-add when safe
• Inplace matrix-multiplication when safe
• Dead code removal

pub fn compile<S: Shape, D: Dev>(self) -> Result<CompiledGraph<S, T, D>>

Compile this graph and insert device-specific optimizations such as CUDA streams.

Trait Implementations

impl<T: Clone + DType> Clone for Graph<T>

fn clone(&self) -> Graph<T>

Returns a duplicate of the value.

const fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.