pytorch/torch/utils
2023-01-12 16:44:29 +00:00
..
backcompat
benchmark Removing invalid git option when updating submodules (#91132) 2022-12-20 02:17:02 +00:00
bottleneck Fix use-dict-literal lint (#83718) 2022-08-24 00:26:46 +00:00
data Enable xdoctest runner in CI for real this time (#83816) 2022-12-29 05:32:42 +00:00
hipify [ROCm] fix hipify mapping for cuDeviceGet (#90726) 2023-01-06 22:57:44 +00:00
jit
model_dump [fix] MathBits: serialization (#88182) 2022-11-09 17:15:12 +00:00
tensorboard Revert D42257039: Multisect successfully blamed D42257039 for test or build failures (#91548) 2023-01-02 21:08:30 +00:00
__init__.py Reland "add an API for external backends to register custom device names (#86992)" (#87453) 2022-10-21 16:51:36 +00:00
_contextlib.py Reland "Add torch.utils.device_mode" (#91796) 2023-01-09 20:57:12 +00:00
_cpp_extension_versioner.py
_crash_handler.py
_cuda_trace.py Add synchronize hooks (#84427) 2022-09-09 13:56:59 +00:00
_device.py Reland "Add torch.utils.device_mode" (#91796) 2023-01-09 20:57:12 +00:00
_foreach_utils.py [utils][foreach] Add group tensor by device and dtype util (#92014) 2023-01-11 23:37:20 +00:00
_freeze.py
_mode_utils.py [Modes] remove enable and rewrite mode stack (squashed) (#84774) 2022-09-27 01:04:35 +00:00
_python_dispatch.py Revert "[Modes] Add assert that the mode isn't already on the stack (#90770)" 2023-01-12 16:44:29 +00:00
_pytree.py Make nested TreeSpec printing nicer (#46538) (#86546) 2022-10-18 16:50:39 +00:00
_traceback.py Add utility report_compile_source_on_error (#91069) 2023-01-11 22:54:46 +00:00
_zip.py Use multipy.package in multipy/runtime (#111) (#82690) 2022-08-03 19:11:12 +00:00
backend_registration.py Enable xdoctest runner in CI for real this time (#83816) 2022-12-29 05:32:42 +00:00
bundled_inputs.py Deprecate TypedStorage, its derived classes, and all of their public methods (#85303) 2022-11-08 18:11:01 +00:00
checkpoint.py Address feedback from previous PR (#86622) 2022-10-10 18:53:41 +00:00
collect_env.py Sets CUDA_MODULE_LOADING to LAZY when not set by the user (#85692) 2022-10-13 14:03:01 +00:00
cpp_backtrace.py Expose cpp_backtrace to python binding (#84896) 2022-09-27 14:59:08 +00:00
cpp_extension.py Some CMake and CUDA cleanup given recent update to C++17 (#90599) 2022-12-30 11:19:26 +00:00
dlpack.py Optimize __dlpack_device__ performance (#86665) 2022-10-11 19:03:46 +00:00
file_baton.py
hooks.py Allow Module forward-pre and forward hooks to take kwargs (#89389) 2022-11-23 02:43:32 +00:00
mkldnn.py
mobile_optimizer.py Revert "Clean Up MobileOptimizerType Rewrite Flags Public API and Documentation (#91600)" 2023-01-10 21:38:40 +00:00
model_zoo.py
show_pickle.py Add __all__ to torch.utils submodules (#85331) 2022-09-27 14:45:26 +00:00
throughput_benchmark.py More doctest refinements. (#83317) 2022-08-22 20:07:26 +00:00
weak.py Deeply rework WeakIdKeyDictionary (#90825) 2022-12-15 08:43:08 +00:00