AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.LinearStepSM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/LinearStepSM86.alpha

354 lines84 declarations21.5 KiBSHA-256 a13116730702
Download exact fileView at revisionf5034773ad4bb52133

def · lines 148–150

linearStepSM86OutputRecordBytesFor

Full file
the output record: y (4m), loss (4), dW (4mk), W' (4mk)
148def linearStepSM86OutputRecordBytesFor =
149  (lambda unrestricted m : Nat . (lambda unrestricted k : Nat .
150    (naturalAdd (naturalMultiply 4 m) (naturalAdd 4 (naturalMultiply 8 (naturalMultiply m k))))))

The compiler supplied declaration spans and resolved links from this source snapshot. This page does not assert that this file belongs to a checked closure.