pytorch/torch/jit
Kurt Mohler 2bfae07a79 Enable dim=None for torch.mean (#81286)
Part of #79525

This will require coordination with XLA before merging, just like #79881
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81286
Approved by: https://github.com/albanD
2022-07-28 22:34:56 +00:00
..
_passes
mobile
__init__.py Fixed docstring typo (#80879) 2022-07-05 16:20:26 +00:00
_async.py
_builtins.py
_check.py
_dataclass_impls.py Reland "[pytorch][PR] Support dataclasses in TorchScript" take 2 (#74353) (#74353) (#76771) 2022-06-07 21:44:55 +00:00
_decomposition_utils.py
_decompositions.py
_freeze.py Fixing the torch.jit.freeze docs (#81020) 2022-07-07 20:33:26 +00:00
_fuser.py Allow user to disable built-in fuser when using TorchDynamo (#81731) 2022-07-22 21:34:49 +00:00
_ir_utils.py
_logging.py
_monkeytype_config.py
_pickle.py
_recursive.py [JIT] Tweak annotation extraction for py3.10 (#81334) 2022-07-12 17:35:30 +00:00
_script.py [JIT] correct torch.jit.Attribute docs 2022-05-11 01:30:15 +00:00
_serialization.py Support saving extra files in pytorch flat buffer format via python (#77870) 2022-05-19 22:15:17 +00:00
_shape_functions.py Enable dim=None for torch.mean (#81286) 2022-07-28 22:34:56 +00:00
_state.py
_trace.py tracer compare_outputs should compare in cdouble for complex 2022-05-06 06:05:50 +00:00
annotations.py Reland "[pytorch][PR] Support dataclasses in TorchScript" take 2 (#74353) (#74353) (#76771) 2022-06-07 21:44:55 +00:00
frontend.py Use generators with all/any in torch/optim (#78142) 2022-06-24 17:23:45 +00:00
generate_bytecode.py
quantized.py
supported_ops.py
unsupported_tensor_ops.py