pytorch/torch/utils
2023-12-20 21:36:25 +00:00
..
_sympy add support for dynamic shapes in round (#115259) 2023-12-19 15:45:50 +00:00
backcompat
benchmark [BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027) 2023-12-20 19:35:08 +00:00
bottleneck
data [Easy][BE]: remove itertools.accumulate Python 2 shim and apply UFMT (#116192) 2023-12-20 20:36:59 +00:00
hipify [ROCm] add hipblaslt support (#114329) 2023-12-20 19:09:25 +00:00
jit
model_dump
tensorboard Fix missing dependency in torch.utils.tensorboard (#115598) 2023-12-20 21:11:52 +00:00
viz
__init__.py New swap function (#111747) 2023-12-08 18:49:35 +00:00
_config_module.py Revert "[dynamo] guarded config (#111299)" (#115386) 2023-12-11 19:35:42 +00:00
_config_typing.pyi Revert "[dynamo] guarded config (#111299)" (#115386) 2023-12-11 19:35:42 +00:00
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cuda_trace.py
_cxx_pytree.py [pytree] expand tree_map to accept multi-inputs (#115642) 2023-12-14 06:16:42 +00:00
_device.py
_foreach_utils.py
_freeze.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
_mode_utils.py
_python_dispatch.py pre_dispatch aot_export (#115188) 2023-12-20 21:36:25 +00:00
_pytree.py [pytree] expand tree_map to accept multi-inputs (#115642) 2023-12-14 06:16:42 +00:00
_stats.py
_traceback.py
_triton.py
_typing_utils.py Re-enable type checking for distributed_c10d.py (#115223) 2023-12-09 11:07:54 +00:00
_zip.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
backend_registration.py
bundled_inputs.py
checkpoint.py Activation checkpoint and checkpoint_sequential errors if use_reentrant not passed explicitly (#115868) 2023-12-20 15:23:44 +00:00
collect_env.py Give users control over packages in torch.utils.collect_env (#112993) 2023-11-28 22:35:25 +00:00
cpp_backtrace.py
cpp_extension.py [ROCm] add hipblaslt support (#114329) 2023-12-20 19:09:25 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py [BE]: Enable RUFF PERF402 and apply fixes (#115505) 2023-12-20 18:01:24 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
show_pickle.py
throughput_benchmark.py
weak.py [pytorch][PR] introduce WeakHashRef (#115216) 2023-12-07 17:48:11 +00:00