candle-graph 0.10.0

TensorFlow Profiler-style execution graphs for candle-rs (trace-only)
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
//! Post-run trace instrumentation — TensorFlow Profiler-style span API.
//!
//! Emits [`crate::trace`] JSONL (`candle-graph/trace/10`). Candle-free: callers supply
//! timings, shapes, and gradient facts from their own probe binary.

mod selector;
mod session;
mod span;

#[cfg(feature = "candle")]
pub mod candle;

pub use selector::CaptureSelector;
pub use session::{ProfileRun, TraceSession};
pub use span::{
    DeviceIntervalRecord, DeviceMemoryRecord, MemoryRecord, OpRecord, SpanGuard, SpanId, SpanKind,
    TensorRecord,
};