Update on "[docs][ao] Add missing docstrings for quantized_max_pool1d and quantized_max_pool2d"

supriyar · supriyar · commit c5d65f112041 · 2021-08-13T17:20:47.000-07:00
Summary: These functions are part of the native functions namespace as well as the quantized namespace Test Plan: CI Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D30316430](https://our.internmc.facebook.com/intern/diff/D30316430) [ghstack-poisoned]
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -11237,10 +11237,10 @@ def merge_dicts(*dicts):
 
 Arguments:
     input (Tensor): quantized tensor
-    weight (Tensor): tensor that corresponds to the gamma, size C
-    bias (Tensor):  tensor that corresponds to the beta, size C
-    mean (Tensor): mean value in batch normalization, size C
-    var (Tensor): variance value, size C
+    weight (Tensor): float tensor that corresponds to the gamma, size C
+    bias (Tensor):  float tensor that corresponds to the beta, size C
+    mean (Tensor): float mean value in batch normalization, size C
+    var (Tensor): float tensor for variance, size C
     eps (float): a value added to the denominator for numerical stability.
     output_scale (float): output quantized tensor scale
     output_zero_point (int): output quantized tensor zero_point
