Realization.Nvidia.SM86.SoftmaxSM86

1631def softmaxSM86ExtentsFor =
1632  (lambda unrestricted variant : (family SoftmaxSM86Variant) .
1633    (eliminate
1634      SoftmaxSM86Variant
1635      (lambda unrestricted current : (family SoftmaxSM86Variant) . (family SoftmaxSM86Extents))
1636      variant
1637      (branch
1638        SoftmaxSM86Forward
1639        .
1640        (constructor
1641          SoftmaxSM86Extents
1642          SoftmaxSM86ExtentsValue
1643          softmaxSM86N256
1644          softmaxSM86N256
1645          zero
1646          softmaxSM86N256
1647          softmaxSM86N1
1648          softmaxSM86N1
1649          softmaxSM86N256
1650          (constructor SoftmaxSM86InputContract SoftmaxSM86FP32ScoresInput)
1651          (constructor SoftmaxSM86OutputContract SoftmaxSM86FP32ProbabilitiesOutput)
1652          (constructor SoftmaxSM86MaskContract SoftmaxSM86Unmasked)))
1653      (branch
1654        SoftmaxSM86CausalForward
1655        .
1656        (constructor
1657          SoftmaxSM86Extents
1658          SoftmaxSM86ExtentsValue
1659          softmaxSM86N256
1660          softmaxSM86N256
1661          zero
1662          softmaxSM86N256
1663          softmaxSM86N1
1664          softmaxSM86N1
1665          softmaxSM86N256
1666          (constructor SoftmaxSM86InputContract SoftmaxSM86FP32ScoresInput)
1667          (constructor SoftmaxSM86OutputContract SoftmaxSM86FP32ProbabilitiesOutput)
1668          (constructor SoftmaxSM86MaskContract SoftmaxSM86CausalLowEightBitRowMask)))
1669      (branch
1670        SoftmaxSM86CausalForward1024
1671        .
1672        (constructor
1673          SoftmaxSM86Extents
1674          SoftmaxSM86ExtentsValue
1675          softmaxSM86N1024
1676          softmaxSM86N1024
1677          zero
1678          softmaxSM86N1024
1679          softmaxSM86U4Natural
1680          softmaxSM86N1
1681          softmaxSM86N256
1682          (constructor SoftmaxSM86InputContract SoftmaxSM86FP32ScoresInput)
1683          (constructor SoftmaxSM86OutputContract SoftmaxSM86FP32ProbabilitiesOutput)
1684          (constructor SoftmaxSM86MaskContract SoftmaxSM86CausalFullRowMask)))
1685      (branch
1686        SoftmaxSM86CausalBackward1024
1687        .
1688        (constructor
1689          SoftmaxSM86Extents
1690          SoftmaxSM86ExtentsValue
1691          softmaxSM86N1024
1692          softmaxSM86N1024
1693          softmaxSM86N1024
1694          softmaxSM86N1024
1695          softmaxSM86U4Natural
1696          softmaxSM86N1
1697          softmaxSM86N256
1698          (constructor SoftmaxSM86InputContract SoftmaxSM86FP32ProbabilitiesAndAdjointsInput)
1699          (constructor SoftmaxSM86OutputContract SoftmaxSM86FP32ScoreAdjointsOutput)
1700          (constructor SoftmaxSM86MaskContract SoftmaxSM86Unmasked)))
1701      (branch
1702        SoftmaxSM86Backward
1703        .
1704        (constructor
1705          SoftmaxSM86Extents
1706          SoftmaxSM86ExtentsValue
1707          softmaxSM86N256
1708          softmaxSM86N256
1709          softmaxSM86N256
1710          softmaxSM86N256
1711          softmaxSM86N1
1712          softmaxSM86N1
1713          softmaxSM86N256
1714          (constructor SoftmaxSM86InputContract SoftmaxSM86FP32ProbabilitiesAndAdjointsInput)
1715          (constructor SoftmaxSM86OutputContract SoftmaxSM86FP32ScoreAdjointsOutput)
1716          (constructor SoftmaxSM86MaskContract SoftmaxSM86Unmasked)))))