Skip to main content

Module pool

Module pool 

Source
Expand description

Tensor 内存池:优化张量分配性能

通过复用已分配的内存减少分配开销,特别适用于 迭代算法(如 PageRank、GNN 训练)中的临时张量

Structs§

ArenaStats
Arena statistics
ArenaTensor
Arena-allocated tensor wrapper
PoolConfig
Tensor 内存池配置
PoolStats
池统计信息
PooledTensor
池化张量:带有自动回收功能的张量包装器
TensorArena
Tensor Arena allocator using bumpalo
TensorPool
Tensor 内存池