2228def cgPrediction : (family NvidiaLaunchSchedule) =
2229 (cgThen
2230 cgForward
2231 (cgPhase
2232 cgPhaseOutput
2233 (cgFor
2234 coppeliusPredictionRows
2235 (lambda unrestricted index : Nat .
2236 (let unrestricted rowLogits =
2237 (naturalAdd cgInference (naturalMultiply index cgRowBytes))
2238 in
2239 (cgThen
2240 (cgCopy
2241 rowLogits
2242 (naturalAdd cgLogits (naturalMultiply (cgPredictionRow index) cgRowBytes))
2243 cgVocab)
2244 (cgLaunch
2245 cgImageArgmax
2246 1
2247 1
2248 1
2249 (cgPointer
2250 (cgArgument 0)
2251 (cgStaging
2252 (naturalAdd
2253 coppeliusResultPredictionsOffset
2254 (naturalMultiply index coppeliusResultPredictionBytes)))
2255 (cgPointer (cgArgument 1) rowLogits cgNoSlots)))))))))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.