Struct TensorEntry 

pub struct TensorEntry {
    pub id: u64,
    pub uop: RwLock<Arc<UOp>>,
    /* private fields */
}

Entry in the global tensor registry.

Uses RwLock for interior mutability of the UOp during global substitution. The RwLock allows concurrent reads (typical tensor operations) with exclusive writes only during apply_map_to_tensors.

Buffer can be set at construction (input tensors) or later (realized tensors). Uses OnceLock for thread-safe one-time initialization.

Fields

id: u64

Unique tensor ID (stable across UOp updates).

uop: RwLock<Arc<UOp>>

The computation graph (mutable for global substitution).

Implementations

impl TensorEntry

pub fn buffer(&self) -> Option<&Arc<Buffer>>

Get the buffer if materialized.

pub fn set_buffer(&self, buffer: Arc<Buffer>) -> bool

Set the buffer (can only be called once, subsequent calls are no-ops). Returns true if buffer was set, false if already set.