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 1389–1409

softmaxSM86Backward1024Adjoint

Full file
dS = P * (dP - sum): fifteen instructions
1389def softmaxSM86Backward1024Adjoint =
1390  (lambda unrestricted probability : (family SM86Register) .
1391    (lambda unrestricted adjoint : (family SM86Register) .
1392      (lambda unrestricted tail : (family SM86Program) .
1393        (softmaxSM86Next
1394          (constructor
1395            SM86InstructionBody
1396            SM86FloatAdd
1397            adjoint
1398            adjoint
1399            softmaxSM86R7
1400            sm86SafeControl)
1401          (softmaxSM86Next
1402            (constructor
1403              SM86InstructionBody
1404              SM86FloatMultiply
1405              probability
1406              probability
1407              adjoint
1408              sm86SafeControl)
1409            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.