Implement BatchNormGrad as multi-output fusion.
Expand BatchNormGrad into smaller ops and use multi-output fusion to speed them up. RELNOTES: n/a PiperOrigin-RevId: 163903421
Loading
Please sign in to comment
Expand BatchNormGrad into smaller ops and use multi-output fusion to speed them up. RELNOTES: n/a PiperOrigin-RevId: 163903421