Skip to main content

trace

Function trace 

Source
pub fn trace<F>(name: &str, f: F) -> Graph
where F: FnOnce(&Tracer) -> Vec<TracedTensor>,
Expand description

Trace a function into an IR graph.

The closure receives a Tracer and returns output tensors. All operations are recorded (not executed) into the graph.