Source/Packages

Realization.Nvidia.SM86.SoftmaxSM86

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

2,114 lines259 declarations74.8 KiBSHA-256 5aa09b49868d

def · lines 1346–1366

softmaxSM86Backward1024Product

Full file
the products P * dP summed into R6 (seven instructions)
1346def softmaxSM86Backward1024Product =
1347  (lambda unrestricted probability : (family SM86Register) .
1348    (lambda unrestricted adjoint : (family SM86Register) .
1349      (lambda unrestricted tail : (family SM86Program) .
1350        (softmaxSM86Next
1351          (constructor
1352            SM86InstructionBody
1353            SM86FloatMultiply
1354            softmaxSM86R22
1355            probability
1356            adjoint
1357            sm86SafeControl)
1358          (softmaxSM86Next
1359            (constructor
1360              SM86InstructionBody
1361              SM86FloatAdd
1362              softmaxSM86R6
1363              softmaxSM86R6
1364              softmaxSM86R22
1365              sm86SafeControl)
1366            tail)))))

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.