message: Update Batchnorm definition post-review
diff: @@ -266,7 +266,7 @@ REGISTER_OP("FusedBatchNormGrad")
  REGISTER_OP("FusedBatchNormGradV2")
  .Input("y_backprop: T")
  .Input("x: T")
  - .Input("scale: U")
  + .Input("scale: V")
  .Input("reserve_space_1: U")
  .Input("reserve_space_2: U")
  .Output("x_backprop: T")
  @@ -276,6 +276,7 @@ REGISTER_OP("FusedBatchNormGradV2")
  .Output("reserve_space_4: U")
  .Attr("T: {half, bfloat16, float}")
  .Attr("U: {half, bfloat16, float}")
  + .Attr("V: {half, bfloat16, float}")
  .Attr("epsilon: float = 0.0001")
  .Attr(GetConvnetDataFormatAttrString())
  .Attr("is_training: bool = true")
  @@ -284,7 +285,7 @@ REGISTER_OP("FusedBatchNormGradV2")
  REGISTER_OP("FusedBatchNormGradV3")
  .Input("y_backprop: T")
  .Input("x: T")
  - .Input("scale: U")
  + .Input("scale: V")
  .Input("reserve_space_1: U")
  .Input("reserve_space_2: U")
  .Input("reserve_space_3: U")
  @@ -294,7 +295,8 @@ REGISTER_OP("FusedBatchNormGradV3")
  .Output("reserve_space_4: U")
  .Output("reserve_space_5: U")
  .Attr("T: {half, bfloat16, float}")
  - .Attr("U: {half, float}")
  + .Attr("U: {half, bfloat16, float}")
  + .Attr("V: {half, bfloat16, float}")
  .Attr("epsilon: float = 0.0001")
  .Attr(GetConvnetDataFormatAttrString())
  .Attr("is_training: bool = true")
  