Struct HookManager 

pub struct HookManager { /* private fields */ }

Hook manager for coordinating debugging hooks

Implementations

impl HookManager

pub fn new() -> Self

Create a new hook manager

pub fn register_hook(&mut self, config: HookConfig) -> Result<Uuid>

Register a new hook

pub fn register_callback(&mut self, name: String, callback: HookCallback)

Register a custom callback

pub fn remove_hook(&mut self, hook_id: Uuid) -> Option<HookConfig>

Remove a hook

pub fn set_hook_enabled(&mut self, hook_id: Uuid, enabled: bool) -> Result<()>

Enable/disable a specific hook

pub fn set_enabled(&mut self, enabled: bool)

Enable/disable all hooks

pub fn set_step(&mut self, step: usize)

Update global step counter

pub fn execute_hooks<T>( &mut self, layer_name: &str, tensor_data: &[T], tensor_shape: &[usize], is_forward: bool, metadata: Option<HashMap<String, String>>, ) -> Vec<(Uuid, HookResult)>

where T: Clone + 'static,

Execute hooks for a tensor operation

pub fn get_hook(&self, hook_id: Uuid) -> Option<&HookConfig>

Get hook configuration

pub fn get_all_hooks(&self) -> Vec<&HookConfig>

Get all hooks

pub fn get_hook_stats(&self, hook_id: Uuid) -> Option<&HookStats>

Get hook statistics

pub fn get_all_stats(&self) -> Vec<&HookStats>

Get all hook statistics

pub fn clear_hooks(&mut self)

Clear all hooks

pub fn create_tensor_inspection_hook( &mut self, layer_patterns: Vec<String>, ) -> Result<Uuid>

Create a convenient tensor inspection hook

pub fn create_gradient_tracking_hook( &mut self, layer_patterns: Vec<String>, ) -> Result<Uuid>

Create a gradient tracking hook

pub fn create_alert_hook( &mut self, condition: HookCondition, message: String, severity: AlertSeverity, ) -> Result<Uuid>

Create a conditional alert hook

Trait Implementations

impl Debug for HookManager

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

Formats the value using the given formatter.

impl Default for HookManager

fn default() -> Self

Returns the “default value” for a type.