我是 Gemini-2.0-flash-exp 打造的 AI 助手,我的小脑袋瓜可厉害啦,帮你咻咻咻地概括文章重点!✨
本文主要分析了内存布局,针对电压膜(Voltage Membrane)和内核权重(Kernel Weight)两种数据结构,分别给出了其内存总大小计算公式和突发传输大小计算公式。其中,电压膜的总大小为 `Co * Hv * Wv`,突发大小为 `Hk`;内核权重的总大小为 `Co * Ci * Hk * Wk`,突发大小为 `Ci * S * Hk * Wk`。最后,文章还提出了一个混合参数单元(Parameter Unit)的计算公式,即 `(Hv * Wv) / (Hk * Wk) + Ci`,用于评估混合数据结构下的内存需求。
Memory Layout Analysis
Voltage Membrane
$$
Total=Co \times H_v \times W_v
$$
$$
Burst=H_k
$$
Kernel Weight
$$
Total=Co \times Ci \times H_k \times W_k
$$
$$
Burst=Ci \times S \times H_k \times W_k
$$
Mixed
$$
Parameter\ Unit =\frac{H_v \times W_v}{H_k \times W_k} + Ci
$$