pytorch/torch/utils
2024-06-24 13:54:30 +00:00
..
_strobelight
_sympy [Inductor] Emit strided block pointer from ModularIndexing and FloorDiv (#127342) 2024-06-16 07:35:57 +00:00
backcompat
benchmark Remove top lev numpy dependency from fuzzer.py (#128759) 2024-06-16 16:34:12 +00:00
bottleneck
data [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
hipify
jit
model_dump
tensorboard
viz
__init__.py [BE] explicitly export subpackage torch.utils (#128342) 2024-06-13 04:39:16 +00:00
_config_module.py
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py
_device.py
_exposed_in.py
_foreach_utils.py [MPS] Fused Adam & AdamW (#127242) 2024-06-18 19:59:50 +00:00
_freeze.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_python_dispatch.py
_pytree.py
_stats.py
_traceback.py
_triton.py
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py Allow SAC policy_fn to return bool for backward compatibility (#129262) 2024-06-24 13:54:30 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py [cpp_extension][inductor] Fix sleef windows depends. (#128770) 2024-06-17 05:44:34 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py Revert "[cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343)" 2024-06-12 18:47:52 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py