message: Scale can be float/bloat16/half in backprop - shouldn't be restricted for input
  Summary: Remove the restriction on the scale value
  Test Plan: Existing tests
  Reviewers: jakeh, jamesn, georgep
diff: @@ -220,7 +220,7 @@ REGISTER_OP("FusedBatchNormGrad")
  REGISTER_OP("FusedBatchNormGradV2")
  .Input("y_backprop: T")
  .Input("x: T")
  - .Input("scale: float")
  + .Input("scale: U")
  .Input("reserve_space_1: U")
  .Input("reserve_space_2: U")
  .Output("x_backprop: T")
  