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