1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
//! # Graph Analysis Module
//!
//! This module contains analysis passes that process the ASG before execution.
//!
//! ## Available Passes
//!
//! - [`ShapeInference`](shape_inference::ShapeInference): Propagates tensor shapes
//! through the graph, detecting shape mismatches before runtime.
//!
//! ## How It Works
//!
//! Analysis passes traverse the graph and compute/validate metadata:
//!
//! ```text
//! ASG (untyped) -> Shape Inference -> ASG (with shapes/dtypes)
//! ```
//!
//! This enables:
//! - **Early error detection**: Shape mismatches caught before execution
//! - **Memory planning**: Know buffer sizes ahead of time
//! - **Optimization opportunities**: Shape-aware graph transformations
//!
//! ## Example
//!
//! ```ignore
//! use rustyasg::analysis::shape_inference::ShapeInference;
//!
//! let mut graph = context.borrow().main_graph().clone();
//!
//! // Provide initial shapes for inputs
//! let shapes = HashMap::from([
//! ("input".to_string(), (vec![32, 784], DType::F32)),
//! ]);
//!
//! // Run shape inference
//! ShapeInference::run(&mut graph, &shapes)?;
//!
//! // Now all nodes have shape information
//! ```