1 2 3 4 5 6 7 8 9 10 11 12 13
//! Post-run trace instrumentation — TensorFlow Profiler-style span API. //! //! Emits [`crate::trace`] JSONL (`candle-graph/trace/6`). Candle-free: callers supply //! timings, shapes, and gradient facts from their own probe binary. mod session; mod span; #[cfg(feature = "candle")] pub mod candle; pub use session::{ProfileRun, TraceSession}; pub use span::{MemoryRecord, OpRecord, SpanGuard, SpanId, SpanKind, TensorRecord};