---- the backward pass ----
2258def cgLayerNormBackward =
2259 (lambda unrestricted gamma : Nat .
2260 (lambda unrestricted gammaGradient : Nat .
2261 (lambda unrestricted betaGradient : Nat .
2262 (lambda unrestricted x : Nat .
2263 (lambda unrestricted dy : Nat .
2264 (lambda unrestricted statistics : Nat .
2265 (lambda unrestricted dx : Nat .
2266 (cgThen
2267 (cgLaunch
2268 cgImageLayerNormBackward
2269 cgSeq
2270 1
2271 1
2272 (cgPointer
2273 (cgArgument 0)
2274 dx
2275 (cgPointer
2276 (cgArgument 1)
2277 x
2278 (cgPointer
2279 (cgArgument 2)
2280 dy
2281 (cgPointer
2282 (cgArgument 3)
2283 gamma
2284 (cgPointer
2285 (cgArgument 4)
2286 (cgArena cgXHat)
2287 (cgPointer (cgArgument 5) statistics (cgLayerNormScalars cgNoSlots))))))))
2288 (cgLaunch
2289 cgImageLayerNormParameters
2290 cgWidth
2291 1
2292 1
2293 (cgPointer
2294 (cgArgument 0)
2295 gammaGradient
2296 (cgPointer
2297 (cgArgument 1)
2298 dy
2299 (cgPointer
2300 (cgArgument 2)
2301 (cgArena cgXHat)
2302 (cgPointer
2303 (cgArgument 3)
2304 betaGradient
2305 (cgSlot (cgScalar 0) cgInverseWidth 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.