tenflowers-core 0.1.1

Core tensor operations and execution engine for TenfloweRS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! Global registry access: OnceLock-based singleton for shape inference.

use super::registry::ShapeInferenceRegistry;
use std::sync::OnceLock;

static GLOBAL_REGISTRY: OnceLock<ShapeInferenceRegistry> = OnceLock::new();

/// Get the global shape inference registry.
///
/// Initializes the registry on first access (lazy initialization).
pub fn get_registry() -> &'static ShapeInferenceRegistry {
    GLOBAL_REGISTRY.get_or_init(ShapeInferenceRegistry::new)
}

/// Initialize the global registry (called automatically on first access).
///
/// This function can be called explicitly to eagerly initialize the registry
/// during application startup, avoiding first-call latency in hot paths.
pub fn initialize_registry() {
    let _ = get_registry();
}