def fork_bytes

stelliferous@0.0.2.0 · kernel_planning.bend:64

import stelliferous@0.0.2.0/kernel_planning.bend as KernelPlanning

import Base

64# Each fork clones the reordered weights and bias, and splits the input rows.
65def fork_bytes(+work: Work,input: Nat) -> Nat:
66  Work{positions,inner,rows,Owners{input_capacity,bias},geometry,activation} = work
67  Nat.add(weight_bytes(work),Nat.add(input,Nat.mul(4n,U32.to_nat(U32.max(2,bias)))))

Uses (8)

Used by (2)