pytorch/torch/utils
2024-06-27 23:16:24 +00:00
..
_strobelight Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_sympy [Inductor] Emit strided block pointer from ModularIndexing and FloorDiv (#127342) 2024-06-16 07:35:57 +00:00
backcompat Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
benchmark [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
bottleneck Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
data [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
hipify Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
jit Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
model_dump Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
tensorboard Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
viz Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
__init__.py [BE] explicitly export subpackage torch.utils (#128342) 2024-06-13 04:39:16 +00:00
_config_module.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_config_typing.pyi Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_content_store.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_contextlib.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_cpp_extension_versioner.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_cxx_pytree.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
_device.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_exposed_in.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_foreach_utils.py [MPS] Fused Adam & AdamW (#127242) 2024-06-18 19:59:50 +00:00
_freeze.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_get_clean_triton.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_import_utils.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_mode_utils.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_python_dispatch.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_pytree.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
_stats.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_traceback.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_triton.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
_typing_utils.py
_zip.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
backend_registration.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
bundled_inputs.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
checkpoint.py Fix device propagation for checkpointing (#128671) 2024-06-27 17:14:13 +00:00
collect_env.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
cpp_backtrace.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
cpp_extension.py [cpp_extension][inductor] Fix sleef windows depends. (#128770) 2024-06-17 05:44:34 +00:00
deterministic.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
dlpack.py
file_baton.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
flop_counter.py [cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343) 2024-06-26 00:49:18 +00:00
hooks.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
mkldnn.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
mobile_optimizer.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
model_zoo.py
module_tracker.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
show_pickle.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
throughput_benchmark.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
weak.py Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00