Struct ShapeGraph 

pub struct ShapeGraph { /* private fields */ }

Shape graph for inference and optimization

Implementations

impl ShapeGraph

pub fn new() -> Self

Create a new empty shape graph

pub fn add_input(&mut self, shape: Vec<usize>) -> NodeId

Add an input node with a known shape

pub fn reshape(&mut self, input: NodeId, target_shape: Vec<usize>) -> NodeId

Add a reshape operation

pub fn transpose(&mut self, input: NodeId, axes: Vec<usize>) -> NodeId

Add a transpose operation

pub fn broadcast(&mut self, input: NodeId, target_shape: Vec<usize>) -> NodeId

Add a broadcast operation

pub fn concatenate( &mut self, input1: NodeId, input2: NodeId, axis: usize, ) -> NodeId

Add a concatenate operation

pub fn stack(&mut self, input1: NodeId, input2: NodeId, axis: usize) -> NodeId

Add a stack operation

pub fn squeeze(&mut self, input: NodeId, axes: Option<Vec<usize>>) -> NodeId

Add a squeeze operation

pub fn unsqueeze(&mut self, input: NodeId, axes: Vec<usize>) -> NodeId

Add an unsqueeze operation

pub fn flatten( &mut self, input: NodeId, start_dim: usize, end_dim: usize, ) -> NodeId

Add a flatten operation

pub fn get_node(&self, id: NodeId) -> Option<&ShapeNode>

Get a node by ID

pub fn infer_shape(&mut self, node_id: NodeId) -> InferenceResult<Vec<usize>>

Infer the shape for a node

pub fn clear_cache(&mut self)

Clear the inference cache

pub fn topological_sort(&self) -> InferenceResult<Vec<NodeId>>

Get all nodes in topological order

pub fn node_count(&self) -> usize

Count the number of nodes in the graph

Trait Implementations

impl Clone for ShapeGraph

fn clone(&self) -> ShapeGraph

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ShapeGraph

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

Formats the value using the given formatter.

impl Default for ShapeGraph

fn default() -> Self

Returns the “default value” for a type.