[BatchNormGrad] Add shape inference support for BatchNormGrad
This is 3rd of 4 CLs to implement BatchNormGrad. Adding shape inference support for BatchNormGrad helps us detect imcompatible shapes and inference the output shape. RELNOTES: n/a PiperOrigin-RevId: 162361218
Loading
Please sign in to comment