diff --git a/docs/master/_dynamo.html b/docs/master/_dynamo.html index 48f4a8150084..c53a1b935a3e 100644 --- a/docs/master/_dynamo.html +++ b/docs/master/_dynamo.html @@ -237,7 +237,7 @@
 
         return result_capturing_wrapper
 
-    # TODO(voz): Handle kwargs properly?
-    flat_args, in_spec = pytree.tree_flatten(args)
+    flat_args, in_spec = pytree.tree_flatten((args, kwargs))
 
     remove_from_cache(f)
     with patch(f"{__name__}.most_recent_backend", None):
@@ -1133,9 +1132,10 @@ Source code for torch._dynamo.eval_frame
     ).transform()
 
     # Make dynamo graph to have same input/output spec as user code
+    input_strs = [f"orig_arg_{i}" for i in range(len(args))] + list(kwargs.keys())
     new_graph.graph._codegen = _PyTreeCodeGen(
         _PyTreeInfo(
-            [f"orig_arg_{i}" for i in range(len(args))],
+            input_strs,
             in_spec,
             out_spec_traced,
         )
diff --git a/docs/master/_modules/torch/_jit_internal.html b/docs/master/_modules/torch/_jit_internal.html
index 4cf6a758df24..a5dda2ec8a53 100644
--- a/docs/master/_modules/torch/_jit_internal.html
+++ b/docs/master/_modules/torch/_jit_internal.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/_lobpcg.html b/docs/master/_modules/torch/_lobpcg.html
index ad571225f1bc..4e22082beee2 100644
--- a/docs/master/_modules/torch/_lobpcg.html
+++ b/docs/master/_modules/torch/_lobpcg.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/_lowrank.html b/docs/master/_modules/torch/_lowrank.html
index bc00f732e9cd..45ca3c478871 100644
--- a/docs/master/_modules/torch/_lowrank.html
+++ b/docs/master/_modules/torch/_lowrank.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/_tensor.html b/docs/master/_modules/torch/_tensor.html
index 7d6880c9fdea..ca4bbc4c9302 100644
--- a/docs/master/_modules/torch/_tensor.html
+++ b/docs/master/_modules/torch/_tensor.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/_tensor_str.html b/docs/master/_modules/torch/_tensor_str.html
index fdf6414d344f..d23d49c24e0b 100644
--- a/docs/master/_modules/torch/_tensor_str.html
+++ b/docs/master/_modules/torch/_tensor_str.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/_utils.html b/docs/master/_modules/torch/_utils.html
index 854d454c4fdc..fdee8203cc70 100644
--- a/docs/master/_modules/torch/_utils.html
+++ b/docs/master/_modules/torch/_utils.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/amp/autocast_mode.html b/docs/master/_modules/torch/amp/autocast_mode.html
index 41282c46a00c..df34abd62d33 100644
--- a/docs/master/_modules/torch/amp/autocast_mode.html
+++ b/docs/master/_modules/torch/amp/autocast_mode.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/modules/fused.html b/docs/master/_modules/torch/ao/nn/intrinsic/modules/fused.html
index c6fb0a8f1e68..275226224e56 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/modules/fused.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/modules/fused.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html b/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html
index d14a52ea1357..f9555599835e 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html b/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html
index 5ed33df3aace..b82518793e3f 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html
index 09e5e38df92b..62024e87840a 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html
index 3685ad1b3645..a924be579725 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html
index 8da8900271eb..d54b199ef072 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html
index 62347ac10f90..15775e1b54ff 100644
--- a/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html
+++ b/docs/master/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/qat/dynamic/modules/linear.html b/docs/master/_modules/torch/ao/nn/qat/dynamic/modules/linear.html
index b723f7d9e1e2..ee7e0d6d1e10 100644
--- a/docs/master/_modules/torch/ao/nn/qat/dynamic/modules/linear.html
+++ b/docs/master/_modules/torch/ao/nn/qat/dynamic/modules/linear.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/qat/modules/conv.html b/docs/master/_modules/torch/ao/nn/qat/modules/conv.html
index c433a64cfd87..0f5b0ba42ec6 100644
--- a/docs/master/_modules/torch/ao/nn/qat/modules/conv.html
+++ b/docs/master/_modules/torch/ao/nn/qat/modules/conv.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/qat/modules/linear.html b/docs/master/_modules/torch/ao/nn/qat/modules/linear.html
index 25738946f019..06d506900fd9 100644
--- a/docs/master/_modules/torch/ao/nn/qat/modules/linear.html
+++ b/docs/master/_modules/torch/ao/nn/qat/modules/linear.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantizable/modules/activation.html b/docs/master/_modules/torch/ao/nn/quantizable/modules/activation.html
index 729504852e6b..63282b4193c3 100644
--- a/docs/master/_modules/torch/ao/nn/quantizable/modules/activation.html
+++ b/docs/master/_modules/torch/ao/nn/quantizable/modules/activation.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantizable/modules/rnn.html b/docs/master/_modules/torch/ao/nn/quantizable/modules/rnn.html
index 4b8d18fe12b7..65038502861e 100644
--- a/docs/master/_modules/torch/ao/nn/quantizable/modules/rnn.html
+++ b/docs/master/_modules/torch/ao/nn/quantizable/modules/rnn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/linear.html b/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/linear.html
index 8affb7bfb189..6548786a8181 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/linear.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/linear.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/rnn.html b/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/rnn.html
index a2e81802d90c..6f354d0207cb 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/rnn.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/dynamic/modules/rnn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/functional.html b/docs/master/_modules/torch/ao/nn/quantized/functional.html
index ae5c7ba215cb..76da1f770cfb 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/functional.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/functional.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/activation.html b/docs/master/_modules/torch/ao/nn/quantized/modules/activation.html
index e06fd5b5376c..3023b403db9c 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/activation.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/activation.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/batchnorm.html b/docs/master/_modules/torch/ao/nn/quantized/modules/batchnorm.html
index 17968ef6e87f..672fad27e7b3 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/batchnorm.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/batchnorm.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/conv.html b/docs/master/_modules/torch/ao/nn/quantized/modules/conv.html
index 35c984c03e24..1a3fefcd0267 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/conv.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/conv.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/embedding_ops.html b/docs/master/_modules/torch/ao/nn/quantized/modules/embedding_ops.html
index 88ea0db67edc..5255c52da5fa 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/embedding_ops.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/embedding_ops.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/functional_modules.html b/docs/master/_modules/torch/ao/nn/quantized/modules/functional_modules.html
index 1afee63c1f55..7b69e40f125c 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/functional_modules.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/functional_modules.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/linear.html b/docs/master/_modules/torch/ao/nn/quantized/modules/linear.html
index 0b32d2c0321f..2c908baa6f80 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/linear.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/linear.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/nn/quantized/modules/normalization.html b/docs/master/_modules/torch/ao/nn/quantized/modules/normalization.html
index 9b9852b7679b..1e326b560a6a 100644
--- a/docs/master/_modules/torch/ao/nn/quantized/modules/normalization.html
+++ b/docs/master/_modules/torch/ao/nn/quantized/modules/normalization.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/ns/_numeric_suite.html b/docs/master/_modules/torch/ao/ns/_numeric_suite.html
index cfedc24d11fa..c554a7c59cd1 100644
--- a/docs/master/_modules/torch/ao/ns/_numeric_suite.html
+++ b/docs/master/_modules/torch/ao/ns/_numeric_suite.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/ns/_numeric_suite_fx.html b/docs/master/_modules/torch/ao/ns/_numeric_suite_fx.html
index fd171918073a..5c7ce95bd5df 100644
--- a/docs/master/_modules/torch/ao/ns/_numeric_suite_fx.html
+++ b/docs/master/_modules/torch/ao/ns/_numeric_suite_fx.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/ns/fx/utils.html b/docs/master/_modules/torch/ao/ns/fx/utils.html
index 98cb67fe4768..61f4da1abdae 100644
--- a/docs/master/_modules/torch/ao/ns/fx/utils.html
+++ b/docs/master/_modules/torch/ao/ns/fx/utils.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/backend_config/backend_config.html b/docs/master/_modules/torch/ao/quantization/backend_config/backend_config.html
index 9b6094b28dc4..973173e9daf5 100644
--- a/docs/master/_modules/torch/ao/quantization/backend_config/backend_config.html
+++ b/docs/master/_modules/torch/ao/quantization/backend_config/backend_config.html
@@ -235,7 +235,7 @@
           
             
     
     
 
@@ -507,7 +507,6 @@ Source code for torch.ao.quantization.backend_config.backend_config
EXTRA_INPUTS_GETTER_DICT_KEY = "extra_inputs_getter"
 NUM_TENSOR_ARGS_TO_OBSERVATION_TYPE_DICT_KEY = "num_tensor_args_to_observation_type"
 INPUT_TYPE_TO_INDEX_DICT_KEY = "input_type_to_index"
-INPUT_OUTPUT_OBSERVED_DICT_KEY = "input_output_observed"
 
 
 # TODO: maybe rename this to something that's not related to observer
@@ -527,6 +526,11 @@ Source code for torch.ao.quantization.backend_config.backend_config
"""this means the output will use the same observer instance as input, based
     on qconfig.activation
     example: torch.cat, maxpool
+    """
+
+    INPUT_OUTPUT_NOT_OBSERVED = 2
+    """this means the input and output are never observed
+    example: x.shape, x.size
     """
 
 
@@ -883,7 +887,6 @@ Source code for torch.ao.quantization.backend_config.backend_config
self._extra_inputs_getter: Optional[Callable] = None
         self._num_tensor_args_to_observation_type: Dict[int, ObservationType] = {}
         self._input_type_to_index: Dict[str, int] = {}
-        self._input_output_observed: Optional[bool] = None
         self._pattern_complex_format: Optional[Pattern] = None
 
     def __repr__(self):
@@ -1024,10 +1027,6 @@ Source code for torch.ao.quantization.backend_config.backend_config
self._input_type_to_index = input_type_to_index
         return self
 
-    def _set_input_output_observed(self, input_output_observed: bool) -> BackendPatternConfig:
-        self._input_output_observed = input_output_observed
-        return self
-
     def _set_pattern_complex_format(self, pattern: Pattern) -> BackendPatternConfig:
         """
         Set the pattern to configure, using the reversed nested tuple format.
@@ -1086,7 +1085,6 @@ Source code for torch.ao.quantization.backend_config.backend_config
conf._set_num_tensor_args_to_observation_type(
             backend_pattern_config_dict.get(NUM_TENSOR_ARGS_TO_OBSERVATION_TYPE_DICT_KEY, {}))
         conf._set_input_type_to_index(backend_pattern_config_dict.get(INPUT_TYPE_TO_INDEX_DICT_KEY, {}))
-        conf._set_input_output_observed(backend_pattern_config_dict.get(INPUT_OUTPUT_OBSERVED_DICT_KEY, None))
         if PATTERN_COMPLEX_FORMAT_DICT_KEY in backend_pattern_config_dict:
             conf._set_pattern_complex_format(backend_pattern_config_dict[PATTERN_COMPLEX_FORMAT_DICT_KEY])
         return conf
@@ -1120,8 +1118,6 @@ Source code for torch.ao.quantization.backend_config.backend_config
backend_pattern_config_dict[NUM_TENSOR_ARGS_TO_OBSERVATION_TYPE_DICT_KEY] = self._num_tensor_args_to_observation_type
         if len(self._input_type_to_index) > 0:
             backend_pattern_config_dict[INPUT_TYPE_TO_INDEX_DICT_KEY] = self._input_type_to_index
-        if self._input_output_observed is not None:
-            backend_pattern_config_dict[INPUT_OUTPUT_OBSERVED_DICT_KEY] = self._input_output_observed
         if self._pattern_complex_format is not None:
             backend_pattern_config_dict[PATTERN_COMPLEX_FORMAT_DICT_KEY] = self._pattern_complex_format
         return backend_pattern_config_dict
diff --git a/docs/master/_modules/torch/ao/quantization/fake_quantize.html b/docs/master/_modules/torch/ao/quantization/fake_quantize.html
index 662e4b18ab37..1e76b81939c5 100644
--- a/docs/master/_modules/torch/ao/quantization/fake_quantize.html
+++ b/docs/master/_modules/torch/ao/quantization/fake_quantize.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/fuse_modules.html b/docs/master/_modules/torch/ao/quantization/fuse_modules.html
index 0c6be05052ee..267448fff5b5 100644
--- a/docs/master/_modules/torch/ao/quantization/fuse_modules.html
+++ b/docs/master/_modules/torch/ao/quantization/fuse_modules.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/fx/custom_config.html b/docs/master/_modules/torch/ao/quantization/fx/custom_config.html
index 06b7666045d8..074b9a5326a3 100644
--- a/docs/master/_modules/torch/ao/quantization/fx/custom_config.html
+++ b/docs/master/_modules/torch/ao/quantization/fx/custom_config.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/observer.html b/docs/master/_modules/torch/ao/quantization/observer.html
index 36784d8d122b..69bf432966e0 100644
--- a/docs/master/_modules/torch/ao/quantization/observer.html
+++ b/docs/master/_modules/torch/ao/quantization/observer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/qconfig.html b/docs/master/_modules/torch/ao/quantization/qconfig.html
index 4287b0aca972..b91728911c05 100644
--- a/docs/master/_modules/torch/ao/quantization/qconfig.html
+++ b/docs/master/_modules/torch/ao/quantization/qconfig.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/qconfig_mapping.html b/docs/master/_modules/torch/ao/quantization/qconfig_mapping.html
index d4bc0e7bebb8..f257f2f3d44d 100644
--- a/docs/master/_modules/torch/ao/quantization/qconfig_mapping.html
+++ b/docs/master/_modules/torch/ao/quantization/qconfig_mapping.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/quantize.html b/docs/master/_modules/torch/ao/quantization/quantize.html
index 5e1b499f7b6a..1b48790b1e67 100644
--- a/docs/master/_modules/torch/ao/quantization/quantize.html
+++ b/docs/master/_modules/torch/ao/quantization/quantize.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/quantize_fx.html b/docs/master/_modules/torch/ao/quantization/quantize_fx.html
index 3a078b08e4b5..1218cbe8391d 100644
--- a/docs/master/_modules/torch/ao/quantization/quantize_fx.html
+++ b/docs/master/_modules/torch/ao/quantization/quantize_fx.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/ao/quantization/stubs.html b/docs/master/_modules/torch/ao/quantization/stubs.html
index 2e639d646fdf..fac915641ce5 100644
--- a/docs/master/_modules/torch/ao/quantization/stubs.html
+++ b/docs/master/_modules/torch/ao/quantization/stubs.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd.html b/docs/master/_modules/torch/autograd.html
index 5d8cb13fef60..88e2a1d251d6 100644
--- a/docs/master/_modules/torch/autograd.html
+++ b/docs/master/_modules/torch/autograd.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/anomaly_mode.html b/docs/master/_modules/torch/autograd/anomaly_mode.html
index 2ee8bb60eb71..4bba01f8f9cc 100644
--- a/docs/master/_modules/torch/autograd/anomaly_mode.html
+++ b/docs/master/_modules/torch/autograd/anomaly_mode.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/forward_ad.html b/docs/master/_modules/torch/autograd/forward_ad.html
index 333679e1311f..40523fe6af20 100644
--- a/docs/master/_modules/torch/autograd/forward_ad.html
+++ b/docs/master/_modules/torch/autograd/forward_ad.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/function.html b/docs/master/_modules/torch/autograd/function.html
index 051a4bf16cf3..814bbd0fb7e6 100644
--- a/docs/master/_modules/torch/autograd/function.html
+++ b/docs/master/_modules/torch/autograd/function.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/functional.html b/docs/master/_modules/torch/autograd/functional.html
index 55405f8e9f68..1785935d2179 100644
--- a/docs/master/_modules/torch/autograd/functional.html
+++ b/docs/master/_modules/torch/autograd/functional.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/grad_mode.html b/docs/master/_modules/torch/autograd/grad_mode.html
index af48aac55966..8bda497f5011 100644
--- a/docs/master/_modules/torch/autograd/grad_mode.html
+++ b/docs/master/_modules/torch/autograd/grad_mode.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/gradcheck.html b/docs/master/_modules/torch/autograd/gradcheck.html
index d26f64653c6e..644450aeb706 100644
--- a/docs/master/_modules/torch/autograd/gradcheck.html
+++ b/docs/master/_modules/torch/autograd/gradcheck.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/graph.html b/docs/master/_modules/torch/autograd/graph.html
index 183647296d42..06f4e5edca12 100644
--- a/docs/master/_modules/torch/autograd/graph.html
+++ b/docs/master/_modules/torch/autograd/graph.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/autograd/profiler.html b/docs/master/_modules/torch/autograd/profiler.html
index 1695c9764574..a589cfe73045 100644
--- a/docs/master/_modules/torch/autograd/profiler.html
+++ b/docs/master/_modules/torch/autograd/profiler.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/cuda.html b/docs/master/_modules/torch/backends/cuda.html
index 2596e511e83f..f29b409e44e7 100644
--- a/docs/master/_modules/torch/backends/cuda.html
+++ b/docs/master/_modules/torch/backends/cuda.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/cudnn.html b/docs/master/_modules/torch/backends/cudnn.html
index 2c32b3a3d532..bc3691ecd0b0 100644
--- a/docs/master/_modules/torch/backends/cudnn.html
+++ b/docs/master/_modules/torch/backends/cudnn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/mkl.html b/docs/master/_modules/torch/backends/mkl.html
index 4304d6040ddf..af60d6f3097e 100644
--- a/docs/master/_modules/torch/backends/mkl.html
+++ b/docs/master/_modules/torch/backends/mkl.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/mkldnn.html b/docs/master/_modules/torch/backends/mkldnn.html
index 82e02c03d3a1..f5f7b15f780c 100644
--- a/docs/master/_modules/torch/backends/mkldnn.html
+++ b/docs/master/_modules/torch/backends/mkldnn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/mps.html b/docs/master/_modules/torch/backends/mps.html
index e88495c44402..61b5b7f4f325 100644
--- a/docs/master/_modules/torch/backends/mps.html
+++ b/docs/master/_modules/torch/backends/mps.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/openmp.html b/docs/master/_modules/torch/backends/openmp.html
index 90f31113898d..407936c69552 100644
--- a/docs/master/_modules/torch/backends/openmp.html
+++ b/docs/master/_modules/torch/backends/openmp.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/backends/opt_einsum.html b/docs/master/_modules/torch/backends/opt_einsum.html
index cebb06c615c5..eebdacf7b9c0 100644
--- a/docs/master/_modules/torch/backends/opt_einsum.html
+++ b/docs/master/_modules/torch/backends/opt_einsum.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cpu/amp/autocast_mode.html b/docs/master/_modules/torch/cpu/amp/autocast_mode.html
index cddbc6bf25f2..e2403e01be4a 100644
--- a/docs/master/_modules/torch/cpu/amp/autocast_mode.html
+++ b/docs/master/_modules/torch/cpu/amp/autocast_mode.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda.html b/docs/master/_modules/torch/cuda.html
index 5a9d584b0a6d..97fbe4fbdb6e 100644
--- a/docs/master/_modules/torch/cuda.html
+++ b/docs/master/_modules/torch/cuda.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/_sanitizer.html b/docs/master/_modules/torch/cuda/_sanitizer.html
index 25a3c9e41e63..5e68f220310b 100644
--- a/docs/master/_modules/torch/cuda/_sanitizer.html
+++ b/docs/master/_modules/torch/cuda/_sanitizer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/amp/autocast_mode.html b/docs/master/_modules/torch/cuda/amp/autocast_mode.html
index a35b7152360c..61d322061153 100644
--- a/docs/master/_modules/torch/cuda/amp/autocast_mode.html
+++ b/docs/master/_modules/torch/cuda/amp/autocast_mode.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/amp/grad_scaler.html b/docs/master/_modules/torch/cuda/amp/grad_scaler.html
index 746b6c3d3054..b356ae221f42 100644
--- a/docs/master/_modules/torch/cuda/amp/grad_scaler.html
+++ b/docs/master/_modules/torch/cuda/amp/grad_scaler.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/graphs.html b/docs/master/_modules/torch/cuda/graphs.html
index d240d62d1494..8b9f809e7c1b 100644
--- a/docs/master/_modules/torch/cuda/graphs.html
+++ b/docs/master/_modules/torch/cuda/graphs.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/jiterator.html b/docs/master/_modules/torch/cuda/jiterator.html
index 25befebbb083..612b4ac993ba 100644
--- a/docs/master/_modules/torch/cuda/jiterator.html
+++ b/docs/master/_modules/torch/cuda/jiterator.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/memory.html b/docs/master/_modules/torch/cuda/memory.html
index c869f4a42729..5fc166e89ff8 100644
--- a/docs/master/_modules/torch/cuda/memory.html
+++ b/docs/master/_modules/torch/cuda/memory.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/nvtx.html b/docs/master/_modules/torch/cuda/nvtx.html
index 056c9e0fc768..fe1c55f4bea2 100644
--- a/docs/master/_modules/torch/cuda/nvtx.html
+++ b/docs/master/_modules/torch/cuda/nvtx.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/random.html b/docs/master/_modules/torch/cuda/random.html
index 8f91981e2d66..d94ab4eea67a 100644
--- a/docs/master/_modules/torch/cuda/random.html
+++ b/docs/master/_modules/torch/cuda/random.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/cuda/streams.html b/docs/master/_modules/torch/cuda/streams.html
index 63dc3be3b627..c76fb83e06e7 100644
--- a/docs/master/_modules/torch/cuda/streams.html
+++ b/docs/master/_modules/torch/cuda/streams.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed.html b/docs/master/_modules/torch/distributed.html
index 34abe08c3a33..d420f704922a 100644
--- a/docs/master/_modules/torch/distributed.html
+++ b/docs/master/_modules/torch/distributed.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.html b/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.html
index 877694cb3d18..90f0126d529b 100644
--- a/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.html
+++ b/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/default_hooks.html b/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/default_hooks.html
index 0f94da0c0cc4..a8ec1ef0b2f1 100644
--- a/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/default_hooks.html
+++ b/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/default_hooks.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.html b/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.html
index a371adbf21a4..db608f158157 100644
--- a/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.html
+++ b/docs/master/_modules/torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/algorithms/join.html b/docs/master/_modules/torch/distributed/algorithms/join.html
index 817e5ef858f7..3f0acb9f88ba 100644
--- a/docs/master/_modules/torch/distributed/algorithms/join.html
+++ b/docs/master/_modules/torch/distributed/algorithms/join.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/autograd.html b/docs/master/_modules/torch/distributed/autograd.html
index 616bb039f2ef..40a2c442f977 100644
--- a/docs/master/_modules/torch/distributed/autograd.html
+++ b/docs/master/_modules/torch/distributed/autograd.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/checkpoint/default_planner.html b/docs/master/_modules/torch/distributed/checkpoint/default_planner.html
index 2f10cb457d01..499655fe1756 100644
--- a/docs/master/_modules/torch/distributed/checkpoint/default_planner.html
+++ b/docs/master/_modules/torch/distributed/checkpoint/default_planner.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/checkpoint/filesystem.html b/docs/master/_modules/torch/distributed/checkpoint/filesystem.html
index 26d92618e595..d8020711a727 100644
--- a/docs/master/_modules/torch/distributed/checkpoint/filesystem.html
+++ b/docs/master/_modules/torch/distributed/checkpoint/filesystem.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/checkpoint/planner.html b/docs/master/_modules/torch/distributed/checkpoint/planner.html
index e6944b190455..1bc9b1895a85 100644
--- a/docs/master/_modules/torch/distributed/checkpoint/planner.html
+++ b/docs/master/_modules/torch/distributed/checkpoint/planner.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/checkpoint/state_dict_loader.html b/docs/master/_modules/torch/distributed/checkpoint/state_dict_loader.html
index e502b5995657..daac422fea04 100644
--- a/docs/master/_modules/torch/distributed/checkpoint/state_dict_loader.html
+++ b/docs/master/_modules/torch/distributed/checkpoint/state_dict_loader.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/checkpoint/state_dict_saver.html b/docs/master/_modules/torch/distributed/checkpoint/state_dict_saver.html
index 5e1d9b972881..bb0fa951b199 100644
--- a/docs/master/_modules/torch/distributed/checkpoint/state_dict_saver.html
+++ b/docs/master/_modules/torch/distributed/checkpoint/state_dict_saver.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/checkpoint/storage.html b/docs/master/_modules/torch/distributed/checkpoint/storage.html
index 5f3437af4f45..6f3fff97075e 100644
--- a/docs/master/_modules/torch/distributed/checkpoint/storage.html
+++ b/docs/master/_modules/torch/distributed/checkpoint/storage.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/distributed_c10d.html b/docs/master/_modules/torch/distributed/distributed_c10d.html
index 9dfe09e67905..bd184b88c59a 100644
--- a/docs/master/_modules/torch/distributed/distributed_c10d.html
+++ b/docs/master/_modules/torch/distributed/distributed_c10d.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/agent/server/api.html b/docs/master/_modules/torch/distributed/elastic/agent/server/api.html
index 04114f75a65c..946b67108f95 100644
--- a/docs/master/_modules/torch/distributed/elastic/agent/server/api.html
+++ b/docs/master/_modules/torch/distributed/elastic/agent/server/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/agent/server/local_elastic_agent.html b/docs/master/_modules/torch/distributed/elastic/agent/server/local_elastic_agent.html
index 831776290aab..1dc7df9a9ff6 100644
--- a/docs/master/_modules/torch/distributed/elastic/agent/server/local_elastic_agent.html
+++ b/docs/master/_modules/torch/distributed/elastic/agent/server/local_elastic_agent.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/events.html b/docs/master/_modules/torch/distributed/elastic/events.html
index fa6e80cb4daf..0cbd0d34c8ee 100644
--- a/docs/master/_modules/torch/distributed/elastic/events.html
+++ b/docs/master/_modules/torch/distributed/elastic/events.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/events/api.html b/docs/master/_modules/torch/distributed/elastic/events/api.html
index f259c34b4a5c..7a7427e1af86 100644
--- a/docs/master/_modules/torch/distributed/elastic/events/api.html
+++ b/docs/master/_modules/torch/distributed/elastic/events/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/events/handlers.html b/docs/master/_modules/torch/distributed/elastic/events/handlers.html
index b3f13f051418..e2d6ff58f6f3 100644
--- a/docs/master/_modules/torch/distributed/elastic/events/handlers.html
+++ b/docs/master/_modules/torch/distributed/elastic/events/handlers.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/metrics/api.html b/docs/master/_modules/torch/distributed/elastic/metrics/api.html
index 0edef2e76420..36c885b59d10 100644
--- a/docs/master/_modules/torch/distributed/elastic/metrics/api.html
+++ b/docs/master/_modules/torch/distributed/elastic/metrics/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/multiprocessing.html b/docs/master/_modules/torch/distributed/elastic/multiprocessing.html
index 84ae45355e89..08ec12a9b583 100644
--- a/docs/master/_modules/torch/distributed/elastic/multiprocessing.html
+++ b/docs/master/_modules/torch/distributed/elastic/multiprocessing.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/multiprocessing/api.html b/docs/master/_modules/torch/distributed/elastic/multiprocessing/api.html
index e617bcbad7fb..92e878c00dac 100644
--- a/docs/master/_modules/torch/distributed/elastic/multiprocessing/api.html
+++ b/docs/master/_modules/torch/distributed/elastic/multiprocessing/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors.html b/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors.html
index e32e7796b5a6..f541b65f9d1b 100644
--- a/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors.html
+++ b/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors/error_handler.html b/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors/error_handler.html
index 50729f6199b6..29dd43881ef1 100644
--- a/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors/error_handler.html
+++ b/docs/master/_modules/torch/distributed/elastic/multiprocessing/errors/error_handler.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/api.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/api.html
index b0163cf15cc4..b23f6013ca5a 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/api.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.html
index ec717c644a93..a3eb69ce5ec2 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/dynamic_rendezvous.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/dynamic_rendezvous.html
index c2f16d66f835..5565398ee598 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/dynamic_rendezvous.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/dynamic_rendezvous.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous.html
index 7ea3bb9cb36c..b135b21f2960 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.html
index fe2e5bdffdcf..872630fbf830 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_server.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_server.html
index 8caaa8b5b412..19c3fe400381 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_server.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_server.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_store.html b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_store.html
index 481802e2f70a..e3ff2784674d 100644
--- a/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_store.html
+++ b/docs/master/_modules/torch/distributed/elastic/rendezvous/etcd_store.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/timer/api.html b/docs/master/_modules/torch/distributed/elastic/timer/api.html
index b3cb359b6cb2..d13c1e00195d 100644
--- a/docs/master/_modules/torch/distributed/elastic/timer/api.html
+++ b/docs/master/_modules/torch/distributed/elastic/timer/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/timer/file_based_local_timer.html b/docs/master/_modules/torch/distributed/elastic/timer/file_based_local_timer.html
index 7e09160ae9ec..56a50fe18009 100644
--- a/docs/master/_modules/torch/distributed/elastic/timer/file_based_local_timer.html
+++ b/docs/master/_modules/torch/distributed/elastic/timer/file_based_local_timer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/elastic/timer/local_timer.html b/docs/master/_modules/torch/distributed/elastic/timer/local_timer.html
index 9c981a20a0c7..1f44105ab74c 100644
--- a/docs/master/_modules/torch/distributed/elastic/timer/local_timer.html
+++ b/docs/master/_modules/torch/distributed/elastic/timer/local_timer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/fsdp/api.html b/docs/master/_modules/torch/distributed/fsdp/api.html
index 40662152e301..63bfb94a47f0 100644
--- a/docs/master/_modules/torch/distributed/fsdp/api.html
+++ b/docs/master/_modules/torch/distributed/fsdp/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/fsdp/fully_sharded_data_parallel.html b/docs/master/_modules/torch/distributed/fsdp/fully_sharded_data_parallel.html
index 6d082bab910d..a971895fe815 100644
--- a/docs/master/_modules/torch/distributed/fsdp/fully_sharded_data_parallel.html
+++ b/docs/master/_modules/torch/distributed/fsdp/fully_sharded_data_parallel.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/nn/api/remote_module.html b/docs/master/_modules/torch/distributed/nn/api/remote_module.html
index 1b3b7ae20f58..df95ce5e8036 100644
--- a/docs/master/_modules/torch/distributed/nn/api/remote_module.html
+++ b/docs/master/_modules/torch/distributed/nn/api/remote_module.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/optim/optimizer.html b/docs/master/_modules/torch/distributed/optim/optimizer.html
index 2121ea3a245e..6973c2c8cb3a 100644
--- a/docs/master/_modules/torch/distributed/optim/optimizer.html
+++ b/docs/master/_modules/torch/distributed/optim/optimizer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/optim/post_localSGD_optimizer.html b/docs/master/_modules/torch/distributed/optim/post_localSGD_optimizer.html
index 4d8a3224452f..4796b647fbba 100644
--- a/docs/master/_modules/torch/distributed/optim/post_localSGD_optimizer.html
+++ b/docs/master/_modules/torch/distributed/optim/post_localSGD_optimizer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/optim/zero_redundancy_optimizer.html b/docs/master/_modules/torch/distributed/optim/zero_redundancy_optimizer.html
index de7ef7c66ddf..c25480326f28 100644
--- a/docs/master/_modules/torch/distributed/optim/zero_redundancy_optimizer.html
+++ b/docs/master/_modules/torch/distributed/optim/zero_redundancy_optimizer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/pipeline/sync/pipe.html b/docs/master/_modules/torch/distributed/pipeline/sync/pipe.html
index bc34bff27ac5..4951db5fd3ab 100644
--- a/docs/master/_modules/torch/distributed/pipeline/sync/pipe.html
+++ b/docs/master/_modules/torch/distributed/pipeline/sync/pipe.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/pipeline/sync/skip/skippable.html b/docs/master/_modules/torch/distributed/pipeline/sync/skip/skippable.html
index 4e07d855f9e8..ee698c5e877b 100644
--- a/docs/master/_modules/torch/distributed/pipeline/sync/skip/skippable.html
+++ b/docs/master/_modules/torch/distributed/pipeline/sync/skip/skippable.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/rpc.html b/docs/master/_modules/torch/distributed/rpc.html
index 4698a151ceed..a0d016ef4c33 100644
--- a/docs/master/_modules/torch/distributed/rpc.html
+++ b/docs/master/_modules/torch/distributed/rpc.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/rpc/api.html b/docs/master/_modules/torch/distributed/rpc/api.html
index ee73555e95d8..a24f0341ed27 100644
--- a/docs/master/_modules/torch/distributed/rpc/api.html
+++ b/docs/master/_modules/torch/distributed/rpc/api.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/rpc/backend_registry.html b/docs/master/_modules/torch/distributed/rpc/backend_registry.html
index bdecf83f87b4..9f17985914d5 100644
--- a/docs/master/_modules/torch/distributed/rpc/backend_registry.html
+++ b/docs/master/_modules/torch/distributed/rpc/backend_registry.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/rpc/functions.html b/docs/master/_modules/torch/distributed/rpc/functions.html
index e73ab713d8c4..58685de40670 100644
--- a/docs/master/_modules/torch/distributed/rpc/functions.html
+++ b/docs/master/_modules/torch/distributed/rpc/functions.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributed/rpc/options.html b/docs/master/_modules/torch/distributed/rpc/options.html
index f1a69a0351c2..016b22680775 100644
--- a/docs/master/_modules/torch/distributed/rpc/options.html
+++ b/docs/master/_modules/torch/distributed/rpc/options.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/bernoulli.html b/docs/master/_modules/torch/distributions/bernoulli.html
index 76ed4b16820c..b2c31c661a60 100644
--- a/docs/master/_modules/torch/distributions/bernoulli.html
+++ b/docs/master/_modules/torch/distributions/bernoulli.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/beta.html b/docs/master/_modules/torch/distributions/beta.html
index 68b732a3a3b8..5adfd1af697f 100644
--- a/docs/master/_modules/torch/distributions/beta.html
+++ b/docs/master/_modules/torch/distributions/beta.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/binomial.html b/docs/master/_modules/torch/distributions/binomial.html
index b8fca48a4ce8..0c8f77be745e 100644
--- a/docs/master/_modules/torch/distributions/binomial.html
+++ b/docs/master/_modules/torch/distributions/binomial.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/categorical.html b/docs/master/_modules/torch/distributions/categorical.html
index 2929b04ff92f..78b5f3694683 100644
--- a/docs/master/_modules/torch/distributions/categorical.html
+++ b/docs/master/_modules/torch/distributions/categorical.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/cauchy.html b/docs/master/_modules/torch/distributions/cauchy.html
index 83148eabd66f..3e0464815588 100644
--- a/docs/master/_modules/torch/distributions/cauchy.html
+++ b/docs/master/_modules/torch/distributions/cauchy.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/chi2.html b/docs/master/_modules/torch/distributions/chi2.html
index 7cd4da0a8557..276fa613c460 100644
--- a/docs/master/_modules/torch/distributions/chi2.html
+++ b/docs/master/_modules/torch/distributions/chi2.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/constraint_registry.html b/docs/master/_modules/torch/distributions/constraint_registry.html
index 86e64cccae21..c4e76f76f135 100644
--- a/docs/master/_modules/torch/distributions/constraint_registry.html
+++ b/docs/master/_modules/torch/distributions/constraint_registry.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/constraints.html b/docs/master/_modules/torch/distributions/constraints.html
index 8b4713ca7abf..48fe04f61f08 100644
--- a/docs/master/_modules/torch/distributions/constraints.html
+++ b/docs/master/_modules/torch/distributions/constraints.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/continuous_bernoulli.html b/docs/master/_modules/torch/distributions/continuous_bernoulli.html
index 4c20f279212b..a2e5e8b04f94 100644
--- a/docs/master/_modules/torch/distributions/continuous_bernoulli.html
+++ b/docs/master/_modules/torch/distributions/continuous_bernoulli.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/dirichlet.html b/docs/master/_modules/torch/distributions/dirichlet.html
index 3469654b6640..8b28cca87c78 100644
--- a/docs/master/_modules/torch/distributions/dirichlet.html
+++ b/docs/master/_modules/torch/distributions/dirichlet.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/distribution.html b/docs/master/_modules/torch/distributions/distribution.html
index c56b6f5d5ef7..d9cc699b5ad7 100644
--- a/docs/master/_modules/torch/distributions/distribution.html
+++ b/docs/master/_modules/torch/distributions/distribution.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/exp_family.html b/docs/master/_modules/torch/distributions/exp_family.html
index 4a248f957bb9..523c4edd2607 100644
--- a/docs/master/_modules/torch/distributions/exp_family.html
+++ b/docs/master/_modules/torch/distributions/exp_family.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/exponential.html b/docs/master/_modules/torch/distributions/exponential.html
index a5bbed750d7d..b5584fdf3d6a 100644
--- a/docs/master/_modules/torch/distributions/exponential.html
+++ b/docs/master/_modules/torch/distributions/exponential.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/fishersnedecor.html b/docs/master/_modules/torch/distributions/fishersnedecor.html
index d3bfe299c331..6a483428a1ca 100644
--- a/docs/master/_modules/torch/distributions/fishersnedecor.html
+++ b/docs/master/_modules/torch/distributions/fishersnedecor.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/gamma.html b/docs/master/_modules/torch/distributions/gamma.html
index 716e90dfdae1..a5a6d286b243 100644
--- a/docs/master/_modules/torch/distributions/gamma.html
+++ b/docs/master/_modules/torch/distributions/gamma.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/geometric.html b/docs/master/_modules/torch/distributions/geometric.html
index 0eacbb85af09..4c4985a2519f 100644
--- a/docs/master/_modules/torch/distributions/geometric.html
+++ b/docs/master/_modules/torch/distributions/geometric.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/gumbel.html b/docs/master/_modules/torch/distributions/gumbel.html
index ebc4f9ede2ab..532b2ee85d95 100644
--- a/docs/master/_modules/torch/distributions/gumbel.html
+++ b/docs/master/_modules/torch/distributions/gumbel.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/half_cauchy.html b/docs/master/_modules/torch/distributions/half_cauchy.html
index e83ce92b9236..c8f0ceb0a50c 100644
--- a/docs/master/_modules/torch/distributions/half_cauchy.html
+++ b/docs/master/_modules/torch/distributions/half_cauchy.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/half_normal.html b/docs/master/_modules/torch/distributions/half_normal.html
index 4de48497eee6..b39f95fb82bf 100644
--- a/docs/master/_modules/torch/distributions/half_normal.html
+++ b/docs/master/_modules/torch/distributions/half_normal.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/independent.html b/docs/master/_modules/torch/distributions/independent.html
index 942c2b8f5781..92e34fc14b2e 100644
--- a/docs/master/_modules/torch/distributions/independent.html
+++ b/docs/master/_modules/torch/distributions/independent.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/kl.html b/docs/master/_modules/torch/distributions/kl.html
index 5247bbe21bcb..0dffbf347f15 100644
--- a/docs/master/_modules/torch/distributions/kl.html
+++ b/docs/master/_modules/torch/distributions/kl.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/kumaraswamy.html b/docs/master/_modules/torch/distributions/kumaraswamy.html
index d6a032b4ca9c..47292c845112 100644
--- a/docs/master/_modules/torch/distributions/kumaraswamy.html
+++ b/docs/master/_modules/torch/distributions/kumaraswamy.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/laplace.html b/docs/master/_modules/torch/distributions/laplace.html
index 47d5bce9fc72..4c8600f02575 100644
--- a/docs/master/_modules/torch/distributions/laplace.html
+++ b/docs/master/_modules/torch/distributions/laplace.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/lkj_cholesky.html b/docs/master/_modules/torch/distributions/lkj_cholesky.html
index 54487d61270c..23fb3c89bbb5 100644
--- a/docs/master/_modules/torch/distributions/lkj_cholesky.html
+++ b/docs/master/_modules/torch/distributions/lkj_cholesky.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/log_normal.html b/docs/master/_modules/torch/distributions/log_normal.html
index effcc29fb9d3..2e4ef2e226ea 100644
--- a/docs/master/_modules/torch/distributions/log_normal.html
+++ b/docs/master/_modules/torch/distributions/log_normal.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/lowrank_multivariate_normal.html b/docs/master/_modules/torch/distributions/lowrank_multivariate_normal.html
index b66def2b6a63..effed32c7771 100644
--- a/docs/master/_modules/torch/distributions/lowrank_multivariate_normal.html
+++ b/docs/master/_modules/torch/distributions/lowrank_multivariate_normal.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/mixture_same_family.html b/docs/master/_modules/torch/distributions/mixture_same_family.html
index 4c93341edb9b..7dc1f04302cb 100644
--- a/docs/master/_modules/torch/distributions/mixture_same_family.html
+++ b/docs/master/_modules/torch/distributions/mixture_same_family.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/multinomial.html b/docs/master/_modules/torch/distributions/multinomial.html
index a4e309373d2b..ac47cee176e8 100644
--- a/docs/master/_modules/torch/distributions/multinomial.html
+++ b/docs/master/_modules/torch/distributions/multinomial.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/multivariate_normal.html b/docs/master/_modules/torch/distributions/multivariate_normal.html
index 3d4cf12f2990..8e009aba721a 100644
--- a/docs/master/_modules/torch/distributions/multivariate_normal.html
+++ b/docs/master/_modules/torch/distributions/multivariate_normal.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/negative_binomial.html b/docs/master/_modules/torch/distributions/negative_binomial.html
index dd4767cbfadd..61d69b470e58 100644
--- a/docs/master/_modules/torch/distributions/negative_binomial.html
+++ b/docs/master/_modules/torch/distributions/negative_binomial.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/normal.html b/docs/master/_modules/torch/distributions/normal.html
index 41e6c6621db0..f242b1baca7c 100644
--- a/docs/master/_modules/torch/distributions/normal.html
+++ b/docs/master/_modules/torch/distributions/normal.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/one_hot_categorical.html b/docs/master/_modules/torch/distributions/one_hot_categorical.html
index 94a4a6b06a0d..51dc1b4e4ca3 100644
--- a/docs/master/_modules/torch/distributions/one_hot_categorical.html
+++ b/docs/master/_modules/torch/distributions/one_hot_categorical.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/pareto.html b/docs/master/_modules/torch/distributions/pareto.html
index 0a492c662cbc..8d9f8d674a02 100644
--- a/docs/master/_modules/torch/distributions/pareto.html
+++ b/docs/master/_modules/torch/distributions/pareto.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/poisson.html b/docs/master/_modules/torch/distributions/poisson.html
index 91c8caac7aa2..41dde3d3ace1 100644
--- a/docs/master/_modules/torch/distributions/poisson.html
+++ b/docs/master/_modules/torch/distributions/poisson.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/relaxed_bernoulli.html b/docs/master/_modules/torch/distributions/relaxed_bernoulli.html
index 736553babb06..abe023b0947c 100644
--- a/docs/master/_modules/torch/distributions/relaxed_bernoulli.html
+++ b/docs/master/_modules/torch/distributions/relaxed_bernoulli.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/relaxed_categorical.html b/docs/master/_modules/torch/distributions/relaxed_categorical.html
index c2d7fc8f679f..f706b1a2c575 100644
--- a/docs/master/_modules/torch/distributions/relaxed_categorical.html
+++ b/docs/master/_modules/torch/distributions/relaxed_categorical.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/studentT.html b/docs/master/_modules/torch/distributions/studentT.html
index 1221ba4cea2e..29cd3a1d49f1 100644
--- a/docs/master/_modules/torch/distributions/studentT.html
+++ b/docs/master/_modules/torch/distributions/studentT.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/transformed_distribution.html b/docs/master/_modules/torch/distributions/transformed_distribution.html
index 13756d6f7305..777e430153fa 100644
--- a/docs/master/_modules/torch/distributions/transformed_distribution.html
+++ b/docs/master/_modules/torch/distributions/transformed_distribution.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/transforms.html b/docs/master/_modules/torch/distributions/transforms.html
index 5b268ab9e57f..de1ed0c5f77e 100644
--- a/docs/master/_modules/torch/distributions/transforms.html
+++ b/docs/master/_modules/torch/distributions/transforms.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/uniform.html b/docs/master/_modules/torch/distributions/uniform.html
index 65216a756c75..f463e6e795bd 100644
--- a/docs/master/_modules/torch/distributions/uniform.html
+++ b/docs/master/_modules/torch/distributions/uniform.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/utils.html b/docs/master/_modules/torch/distributions/utils.html
index fec55448f7ea..6401d3ccf618 100644
--- a/docs/master/_modules/torch/distributions/utils.html
+++ b/docs/master/_modules/torch/distributions/utils.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/von_mises.html b/docs/master/_modules/torch/distributions/von_mises.html
index 15b5f413b90c..6fe199394be6 100644
--- a/docs/master/_modules/torch/distributions/von_mises.html
+++ b/docs/master/_modules/torch/distributions/von_mises.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/weibull.html b/docs/master/_modules/torch/distributions/weibull.html
index a8c357df7cc4..83210fbf1b52 100644
--- a/docs/master/_modules/torch/distributions/weibull.html
+++ b/docs/master/_modules/torch/distributions/weibull.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/distributions/wishart.html b/docs/master/_modules/torch/distributions/wishart.html
index 33a247a08e1a..b6eb3ad9864b 100644
--- a/docs/master/_modules/torch/distributions/wishart.html
+++ b/docs/master/_modules/torch/distributions/wishart.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/func.html b/docs/master/_modules/torch/func.html
index d68787bada2e..3a3717e7eeb9 100644
--- a/docs/master/_modules/torch/func.html
+++ b/docs/master/_modules/torch/func.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/functional.html b/docs/master/_modules/torch/functional.html
index b7769ac9eba5..859944e7c4ad 100644
--- a/docs/master/_modules/torch/functional.html
+++ b/docs/master/_modules/torch/functional.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/futures.html b/docs/master/_modules/torch/futures.html
index 1fc2609af9c4..361b514c34b1 100644
--- a/docs/master/_modules/torch/futures.html
+++ b/docs/master/_modules/torch/futures.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/fx/_symbolic_trace.html b/docs/master/_modules/torch/fx/_symbolic_trace.html
index a7aeff1e9cde..6239dbfec37c 100644
--- a/docs/master/_modules/torch/fx/_symbolic_trace.html
+++ b/docs/master/_modules/torch/fx/_symbolic_trace.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/fx/graph.html b/docs/master/_modules/torch/fx/graph.html
index d35e01fa62db..3380e46fe79d 100644
--- a/docs/master/_modules/torch/fx/graph.html
+++ b/docs/master/_modules/torch/fx/graph.html
@@ -235,7 +235,7 @@
           
             
     
     
 
@@ -1088,17 +1088,49 @@ Source code for torch.fx.graph
         return pytree.tree_unflatten(out, self.pytree_info.out_spec)
 
     def gen_fn_def(self, free_vars, maybe_return_annotation):
+        # Given a user function/model:
+        #   myargs = [myargs0, myargs1]
+        #   mykwargs = {'mykwargs0': ..., 'mykwargs1': ...}
+        #   def forward(self, mypos, *myargs, mykey=None, **mykwargs):
+        #
+        # The generated code flattens all keywords into positional arguments for `forward()`
+        #   e.g forward(self, mypos, myargs0, myargs1, mykey, mykwargs0, mykwargs1):
+        #
+        # Within `forward`, `tree_flatten_spec``still parses args and kwargs separately
+        #   e.g. tree_flatten_spec(([mypos, myargs0, myargs1],
+        #                           {'mykey':mykey, 'mykwargs0':mykwargs0, 'mykwargs1':mykwargs1}),
+        #                          self._in_spec)
+        #
+        # If the user function/model does not have keywords, the dict is suppressed from tree_flatten_spec
+        #   e.g. tree_flatten_spec([mypos, myargs0, myargs1]), self._in_spec)
         if self.pytree_info is None:
             return super().gen_fn_def(free_vars, maybe_return_annotation)
-        function_args = self.pytree_info.orig_args
-        has_orig_self = (function_args[0] == 'self') if len(function_args) > 0 else False
+
+        fn_args = self.pytree_info.orig_args
+        has_orig_self = (fn_args[0] == 'self') if len(fn_args) > 0 else False
         if has_orig_self:
             free_vars.insert(0, 'self')
-        function_definition = super().gen_fn_def(function_args[:], maybe_return_annotation)
+        fn_definition = super().gen_fn_def(fn_args[:], maybe_return_annotation)
+
         if len(free_vars) > 0:  # pytree has placeholders in it
-            function_definition += f"""
-    {', '.join(free_vars)}, = fx_pytree.tree_flatten_spec([{', '.join(function_args)}], self._in_spec)"""
-        return function_definition
+            # when kwargs is present, in_spec is tuple(args, kwargs)
+            has_args_kwargs_tuple = self.pytree_info.in_spec.type == tuple and \
+                len(self.pytree_info.in_spec.children_specs) == 2 and \
+                self.pytree_info.in_spec.children_specs[0].type == tuple and \
+                self.pytree_info.in_spec.children_specs[1].type == dict
+            fn_kwargs = '{}'
+            fn_signature = f"[{', '.join(fn_args)}], self._in_spec"
+            if has_args_kwargs_tuple:
+                count_args = len(self.pytree_info.in_spec.children_specs[0].children_specs)
+                fn_args = self.pytree_info.orig_args[:count_args]
+                fn_kwargs = '{' + ', '.join(f"'{k}':{v}" for k, v in zip(
+                                  self.pytree_info.in_spec.children_specs[1].context,
+                                  self.pytree_info.orig_args[count_args:])) + '}'
+                fn_signature = f"([{', '.join(fn_args)}], {fn_kwargs}), self._in_spec"
+
+            fn_definition += f"""
+    {', '.join(free_vars)}, = fx_pytree.tree_flatten_spec({fn_signature})"""
+        return fn_definition
 
     def generate_output(self, output_args):
         if self.pytree_info:
diff --git a/docs/master/_modules/torch/fx/graph_module.html b/docs/master/_modules/torch/fx/graph_module.html
index 1f8369f63075..0b8dc2352fb5 100644
--- a/docs/master/_modules/torch/fx/graph_module.html
+++ b/docs/master/_modules/torch/fx/graph_module.html
@@ -235,7 +235,7 @@
           
             
     
     
 
@@ -1171,11 +1171,12 @@ Source code for torch.fx.graph_module
     # and cause symbolic tracing to occur every time we try to copy the object
     def __deepcopy__(self, memo):
         fake_mod = torch.nn.Module()
-        fake_mod.__dict__ = copy.deepcopy(self.__dict__)
+        fake_mod.__dict__ = copy.deepcopy(self.__dict__, memo)
         res = GraphModule(fake_mod, fake_mod.__dict__['_graph'])
-        res.meta = copy.deepcopy(getattr(self, 'meta', {}))
+        res.meta = copy.deepcopy(getattr(self, 'meta', {}), memo)
         return res
 
+
     def __copy__(self):
         res = GraphModule(self, self.graph)
         res.meta = getattr(self, 'meta', {})
diff --git a/docs/master/_modules/torch/fx/interpreter.html b/docs/master/_modules/torch/fx/interpreter.html
index b8a292493b55..c34ca2e48a07 100644
--- a/docs/master/_modules/torch/fx/interpreter.html
+++ b/docs/master/_modules/torch/fx/interpreter.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/fx/node.html b/docs/master/_modules/torch/fx/node.html
index fba20564b734..10eaf296c9fa 100644
--- a/docs/master/_modules/torch/fx/node.html
+++ b/docs/master/_modules/torch/fx/node.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/fx/proxy.html b/docs/master/_modules/torch/fx/proxy.html
index 1e58d3f24eaf..e7d64a395401 100644
--- a/docs/master/_modules/torch/fx/proxy.html
+++ b/docs/master/_modules/torch/fx/proxy.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/fx/subgraph_rewriter.html b/docs/master/_modules/torch/fx/subgraph_rewriter.html
index 592c70a4cfe0..a600b18e52ac 100644
--- a/docs/master/_modules/torch/fx/subgraph_rewriter.html
+++ b/docs/master/_modules/torch/fx/subgraph_rewriter.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/hub.html b/docs/master/_modules/torch/hub.html
index d12331c46c42..41eff05551de 100644
--- a/docs/master/_modules/torch/hub.html
+++ b/docs/master/_modules/torch/hub.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit.html b/docs/master/_modules/torch/jit.html
index 6bbaeacd77c2..8cba3174d6a0 100644
--- a/docs/master/_modules/torch/jit.html
+++ b/docs/master/_modules/torch/jit.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit/_async.html b/docs/master/_modules/torch/jit/_async.html
index 590b4b8e71f4..3e319818cff2 100644
--- a/docs/master/_modules/torch/jit/_async.html
+++ b/docs/master/_modules/torch/jit/_async.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit/_freeze.html b/docs/master/_modules/torch/jit/_freeze.html
index caded0d282dd..d19e58d6e360 100644
--- a/docs/master/_modules/torch/jit/_freeze.html
+++ b/docs/master/_modules/torch/jit/_freeze.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit/_fuser.html b/docs/master/_modules/torch/jit/_fuser.html
index 66551359d6dc..7f718fdf2f74 100644
--- a/docs/master/_modules/torch/jit/_fuser.html
+++ b/docs/master/_modules/torch/jit/_fuser.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit/_script.html b/docs/master/_modules/torch/jit/_script.html
index 0687d9c5fbcf..6a7b4201c54f 100644
--- a/docs/master/_modules/torch/jit/_script.html
+++ b/docs/master/_modules/torch/jit/_script.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit/_serialization.html b/docs/master/_modules/torch/jit/_serialization.html
index 47fe287c72b9..5e28f519f5ca 100644
--- a/docs/master/_modules/torch/jit/_serialization.html
+++ b/docs/master/_modules/torch/jit/_serialization.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/jit/_trace.html b/docs/master/_modules/torch/jit/_trace.html
index a4dcdf4c5a01..42d82b91d26b 100644
--- a/docs/master/_modules/torch/jit/_trace.html
+++ b/docs/master/_modules/torch/jit/_trace.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/library.html b/docs/master/_modules/torch/library.html
index 17246f64c33f..0178e6133f57 100644
--- a/docs/master/_modules/torch/library.html
+++ b/docs/master/_modules/torch/library.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/monitor.html b/docs/master/_modules/torch/monitor.html
index 44822f371162..ca4291ee03f1 100644
--- a/docs/master/_modules/torch/monitor.html
+++ b/docs/master/_modules/torch/monitor.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/multiprocessing.html b/docs/master/_modules/torch/multiprocessing.html
index 3e0c58286b95..18cf2a770c81 100644
--- a/docs/master/_modules/torch/multiprocessing.html
+++ b/docs/master/_modules/torch/multiprocessing.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/multiprocessing/spawn.html b/docs/master/_modules/torch/multiprocessing/spawn.html
index a13fa86b1508..55e46ac92189 100644
--- a/docs/master/_modules/torch/multiprocessing/spawn.html
+++ b/docs/master/_modules/torch/multiprocessing/spawn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nested.html b/docs/master/_modules/torch/nested.html
index 0545ae7f17dd..3496bf7c52d5 100644
--- a/docs/master/_modules/torch/nested.html
+++ b/docs/master/_modules/torch/nested.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/functional.html b/docs/master/_modules/torch/nn/functional.html
index 0679e0eec0d1..304ce7b2a6fc 100644
--- a/docs/master/_modules/torch/nn/functional.html
+++ b/docs/master/_modules/torch/nn/functional.html
@@ -235,7 +235,7 @@
           
             
     
     
 
@@ -5328,7 +5328,7 @@ Source code for torch.nn.functional
 def _scaled_dot_product_attention(
         query: Tensor,
         key: Tensor,
-        value,
+        value: Tensor,
         attn_mask: Optional[Tensor] = None,
         dropout_p: float = 0.0,
         need_attn_weights: bool = False,
@@ -5336,7 +5336,7 @@ Source code for torch.nn.functional
     r""" TODO This function is for merge purposes only and needs to be removed
     """
     warnings.warn("This function is deprecated please rebuild your models with the public version of sdpa.")
-    return torch._C._nn.scaled_dot_product_attention(query, key, value, attn_mask, dropout_p, need_attn_weights, is_causal)
+    return torch._C._nn._scaled_dot_product_attention(query, key, value, attn_mask, dropout_p, need_attn_weights, is_causal)
 
 def _mha_shape_check(query: Tensor, key: Tensor, value: Tensor,
                      key_padding_mask: Optional[Tensor], attn_mask: Optional[Tensor], num_heads: int):
@@ -5737,7 +5737,7 @@ Source code for torch.nn.functional
         k = k.view(bsz, num_heads, src_len, head_dim)
         v = v.view(bsz, num_heads, src_len, head_dim)
 
-        attn_output = scaled_dot_product_attention(q, k, v, attn_mask, dropout_p, is_causal)
+        attn_output, _ = _scaled_dot_product_attention(q, k, v, attn_mask, dropout_p, False, is_causal)
         attn_output = attn_output.permute(2, 0, 1, 3).contiguous().view(bsz * tgt_len, embed_dim)
 
         attn_output = linear(attn_output, out_proj_weight, out_proj_bias)
diff --git a/docs/master/_modules/torch/nn/init.html b/docs/master/_modules/torch/nn/init.html
index 1b82f2b2d2de..97edcde7f6f2 100644
--- a/docs/master/_modules/torch/nn/init.html
+++ b/docs/master/_modules/torch/nn/init.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/activation.html b/docs/master/_modules/torch/nn/modules/activation.html
index 1b7c831e4af6..e92b15a4ab1d 100644
--- a/docs/master/_modules/torch/nn/modules/activation.html
+++ b/docs/master/_modules/torch/nn/modules/activation.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/adaptive.html b/docs/master/_modules/torch/nn/modules/adaptive.html
index 3acab41c2144..8564df92b1eb 100644
--- a/docs/master/_modules/torch/nn/modules/adaptive.html
+++ b/docs/master/_modules/torch/nn/modules/adaptive.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/batchnorm.html b/docs/master/_modules/torch/nn/modules/batchnorm.html
index 166ed2850837..9b0d54600451 100644
--- a/docs/master/_modules/torch/nn/modules/batchnorm.html
+++ b/docs/master/_modules/torch/nn/modules/batchnorm.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/channelshuffle.html b/docs/master/_modules/torch/nn/modules/channelshuffle.html
index 7068e3751c10..1ba5498bb80c 100644
--- a/docs/master/_modules/torch/nn/modules/channelshuffle.html
+++ b/docs/master/_modules/torch/nn/modules/channelshuffle.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/container.html b/docs/master/_modules/torch/nn/modules/container.html
index b01276131465..509a2233d2d9 100644
--- a/docs/master/_modules/torch/nn/modules/container.html
+++ b/docs/master/_modules/torch/nn/modules/container.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/conv.html b/docs/master/_modules/torch/nn/modules/conv.html
index 09a70c60a030..c7f50bf59163 100644
--- a/docs/master/_modules/torch/nn/modules/conv.html
+++ b/docs/master/_modules/torch/nn/modules/conv.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/distance.html b/docs/master/_modules/torch/nn/modules/distance.html
index eb11e14336b6..201e1345b4d7 100644
--- a/docs/master/_modules/torch/nn/modules/distance.html
+++ b/docs/master/_modules/torch/nn/modules/distance.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/dropout.html b/docs/master/_modules/torch/nn/modules/dropout.html
index 04625077f1f3..512d33358fb4 100644
--- a/docs/master/_modules/torch/nn/modules/dropout.html
+++ b/docs/master/_modules/torch/nn/modules/dropout.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/flatten.html b/docs/master/_modules/torch/nn/modules/flatten.html
index 38e636d8d1bb..62df093db11f 100644
--- a/docs/master/_modules/torch/nn/modules/flatten.html
+++ b/docs/master/_modules/torch/nn/modules/flatten.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/fold.html b/docs/master/_modules/torch/nn/modules/fold.html
index 344e05331cd7..69d3c4d91a7e 100644
--- a/docs/master/_modules/torch/nn/modules/fold.html
+++ b/docs/master/_modules/torch/nn/modules/fold.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/instancenorm.html b/docs/master/_modules/torch/nn/modules/instancenorm.html
index b61b42015be5..87de7f7f597b 100644
--- a/docs/master/_modules/torch/nn/modules/instancenorm.html
+++ b/docs/master/_modules/torch/nn/modules/instancenorm.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/lazy.html b/docs/master/_modules/torch/nn/modules/lazy.html
index 9726ff38be07..cf6a4391db54 100644
--- a/docs/master/_modules/torch/nn/modules/lazy.html
+++ b/docs/master/_modules/torch/nn/modules/lazy.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/linear.html b/docs/master/_modules/torch/nn/modules/linear.html
index e18f7e9412ab..86a0dfcf6ae1 100644
--- a/docs/master/_modules/torch/nn/modules/linear.html
+++ b/docs/master/_modules/torch/nn/modules/linear.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/loss.html b/docs/master/_modules/torch/nn/modules/loss.html
index 5e7092401f5a..6e965ee61ca5 100644
--- a/docs/master/_modules/torch/nn/modules/loss.html
+++ b/docs/master/_modules/torch/nn/modules/loss.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/module.html b/docs/master/_modules/torch/nn/modules/module.html
index eb53289a9113..749d84a334f6 100644
--- a/docs/master/_modules/torch/nn/modules/module.html
+++ b/docs/master/_modules/torch/nn/modules/module.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/normalization.html b/docs/master/_modules/torch/nn/modules/normalization.html
index 87a73cfc47ec..3c957f05fe6a 100644
--- a/docs/master/_modules/torch/nn/modules/normalization.html
+++ b/docs/master/_modules/torch/nn/modules/normalization.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/padding.html b/docs/master/_modules/torch/nn/modules/padding.html
index 7d9c54e3c3ef..977c00b9bc93 100644
--- a/docs/master/_modules/torch/nn/modules/padding.html
+++ b/docs/master/_modules/torch/nn/modules/padding.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/pixelshuffle.html b/docs/master/_modules/torch/nn/modules/pixelshuffle.html
index 2e7053fcaa53..17f1d9d11c0f 100644
--- a/docs/master/_modules/torch/nn/modules/pixelshuffle.html
+++ b/docs/master/_modules/torch/nn/modules/pixelshuffle.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/pooling.html b/docs/master/_modules/torch/nn/modules/pooling.html
index 3780ae342d82..8273b95d3491 100644
--- a/docs/master/_modules/torch/nn/modules/pooling.html
+++ b/docs/master/_modules/torch/nn/modules/pooling.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/rnn.html b/docs/master/_modules/torch/nn/modules/rnn.html
index d9d9b1056196..534fe49717ac 100644
--- a/docs/master/_modules/torch/nn/modules/rnn.html
+++ b/docs/master/_modules/torch/nn/modules/rnn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/sparse.html b/docs/master/_modules/torch/nn/modules/sparse.html
index 2160ba62d9a7..e00ec3af0829 100644
--- a/docs/master/_modules/torch/nn/modules/sparse.html
+++ b/docs/master/_modules/torch/nn/modules/sparse.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/transformer.html b/docs/master/_modules/torch/nn/modules/transformer.html
index d960555e113b..61effcdb5870 100644
--- a/docs/master/_modules/torch/nn/modules/transformer.html
+++ b/docs/master/_modules/torch/nn/modules/transformer.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/modules/upsampling.html b/docs/master/_modules/torch/nn/modules/upsampling.html
index 9f920ca67861..4895436e1b65 100644
--- a/docs/master/_modules/torch/nn/modules/upsampling.html
+++ b/docs/master/_modules/torch/nn/modules/upsampling.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/parallel/comm.html b/docs/master/_modules/torch/nn/parallel/comm.html
index 21fe69f2e432..f76d19462cbc 100644
--- a/docs/master/_modules/torch/nn/parallel/comm.html
+++ b/docs/master/_modules/torch/nn/parallel/comm.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/parallel/data_parallel.html b/docs/master/_modules/torch/nn/parallel/data_parallel.html
index 8bf9bee0212a..adecc95ee4e0 100644
--- a/docs/master/_modules/torch/nn/parallel/data_parallel.html
+++ b/docs/master/_modules/torch/nn/parallel/data_parallel.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/parallel/distributed.html b/docs/master/_modules/torch/nn/parallel/distributed.html
index 437bfdd1fa49..26867504c887 100644
--- a/docs/master/_modules/torch/nn/parallel/distributed.html
+++ b/docs/master/_modules/torch/nn/parallel/distributed.html
@@ -235,7 +235,7 @@
           
             
     
     
 
@@ -1611,7 +1611,6 @@ Source code for torch.nn.parallel.distributed
# sync params according to location (before/after forward) user
             # specified as part of hook, if hook was specified.
-            buffer_hook_registered = hasattr(self, "buffer_hook")
             if self._check_sync_bufs_pre_fwd():
                 self._sync_buffers()
 
diff --git a/docs/master/_modules/torch/nn/parameter.html b/docs/master/_modules/torch/nn/parameter.html
index 2988976803d6..82aced984b3a 100644
--- a/docs/master/_modules/torch/nn/parameter.html
+++ b/docs/master/_modules/torch/nn/parameter.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/clip_grad.html b/docs/master/_modules/torch/nn/utils/clip_grad.html
index 73a2828c40db..02e39e494435 100644
--- a/docs/master/_modules/torch/nn/utils/clip_grad.html
+++ b/docs/master/_modules/torch/nn/utils/clip_grad.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/convert_parameters.html b/docs/master/_modules/torch/nn/utils/convert_parameters.html
index 6510bb56a5f9..e193b089d75e 100644
--- a/docs/master/_modules/torch/nn/utils/convert_parameters.html
+++ b/docs/master/_modules/torch/nn/utils/convert_parameters.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/init.html b/docs/master/_modules/torch/nn/utils/init.html
index f5288efe4e2a..2637057043f3 100644
--- a/docs/master/_modules/torch/nn/utils/init.html
+++ b/docs/master/_modules/torch/nn/utils/init.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/parametrizations.html b/docs/master/_modules/torch/nn/utils/parametrizations.html
index 5a0a193f4b21..2307d1946a62 100644
--- a/docs/master/_modules/torch/nn/utils/parametrizations.html
+++ b/docs/master/_modules/torch/nn/utils/parametrizations.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/parametrize.html b/docs/master/_modules/torch/nn/utils/parametrize.html
index adc1b18ad6b8..f2627645ad37 100644
--- a/docs/master/_modules/torch/nn/utils/parametrize.html
+++ b/docs/master/_modules/torch/nn/utils/parametrize.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/prune.html b/docs/master/_modules/torch/nn/utils/prune.html
index 3ad0bf4b05d4..677e4daa5eee 100644
--- a/docs/master/_modules/torch/nn/utils/prune.html
+++ b/docs/master/_modules/torch/nn/utils/prune.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/rnn.html b/docs/master/_modules/torch/nn/utils/rnn.html
index 674c044b6376..c359300d025f 100644
--- a/docs/master/_modules/torch/nn/utils/rnn.html
+++ b/docs/master/_modules/torch/nn/utils/rnn.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/spectral_norm.html b/docs/master/_modules/torch/nn/utils/spectral_norm.html
index b0f6291b04f5..1119da2bbf24 100644
--- a/docs/master/_modules/torch/nn/utils/spectral_norm.html
+++ b/docs/master/_modules/torch/nn/utils/spectral_norm.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/stateless.html b/docs/master/_modules/torch/nn/utils/stateless.html
index 7c4ed53dc73e..e6de41f1d7dd 100644
--- a/docs/master/_modules/torch/nn/utils/stateless.html
+++ b/docs/master/_modules/torch/nn/utils/stateless.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/nn/utils/weight_norm.html b/docs/master/_modules/torch/nn/utils/weight_norm.html
index 7d07b96cd20c..fac58d514162 100644
--- a/docs/master/_modules/torch/nn/utils/weight_norm.html
+++ b/docs/master/_modules/torch/nn/utils/weight_norm.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/onnx.html b/docs/master/_modules/torch/onnx.html
index e22c3e50a5a1..439929aa0b51 100644
--- a/docs/master/_modules/torch/onnx.html
+++ b/docs/master/_modules/torch/onnx.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/onnx/_internal/diagnostics/_diagnostic.html b/docs/master/_modules/torch/onnx/_internal/diagnostics/_diagnostic.html
index d11ac753410a..1d6e32c87806 100644
--- a/docs/master/_modules/torch/onnx/_internal/diagnostics/_diagnostic.html
+++ b/docs/master/_modules/torch/onnx/_internal/diagnostics/_diagnostic.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/onnx/_internal/diagnostics/infra/engine.html b/docs/master/_modules/torch/onnx/_internal/diagnostics/infra/engine.html
index 1459a7298dc9..fe62caf34f27 100644
--- a/docs/master/_modules/torch/onnx/_internal/diagnostics/infra/engine.html
+++ b/docs/master/_modules/torch/onnx/_internal/diagnostics/infra/engine.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/onnx/_type_utils.html b/docs/master/_modules/torch/onnx/_type_utils.html
index 0715178454af..4970c4b4e1db 100644
--- a/docs/master/_modules/torch/onnx/_type_utils.html
+++ b/docs/master/_modules/torch/onnx/_type_utils.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/onnx/utils.html b/docs/master/_modules/torch/onnx/utils.html
index c6e33465b6dd..2b2c0f1e8e03 100644
--- a/docs/master/_modules/torch/onnx/utils.html
+++ b/docs/master/_modules/torch/onnx/utils.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/onnx/verification.html b/docs/master/_modules/torch/onnx/verification.html
index a56053d027d4..91795345abb1 100644
--- a/docs/master/_modules/torch/onnx/verification.html
+++ b/docs/master/_modules/torch/onnx/verification.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/optim/adadelta.html b/docs/master/_modules/torch/optim/adadelta.html
index 04150424662a..1a55c4a18f2f 100644
--- a/docs/master/_modules/torch/optim/adadelta.html
+++ b/docs/master/_modules/torch/optim/adadelta.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/optim/adagrad.html b/docs/master/_modules/torch/optim/adagrad.html
index aa6264048c56..69c642107f23 100644
--- a/docs/master/_modules/torch/optim/adagrad.html
+++ b/docs/master/_modules/torch/optim/adagrad.html
@@ -235,7 +235,7 @@
           
             
     
     
 
diff --git a/docs/master/_modules/torch/optim/adam.html b/docs/master/_modules/torch/optim/adam.html
index 52289554697a..9c5aa7ac89c4 100644
--- a/docs/master/_modules/torch/optim/adam.html
+++ b/docs/master/_modules/torch/optim/adam.html
@@ -235,7 +235,7 @@
           
             
     
-