pytorch/torch/utils
2023-09-26 12:10:25 +00:00
..
_sympy Basic fp8 support in Inductor (#109168) 2023-09-23 04:41:41 +00:00
backcompat
benchmark Fix torch.utils.benchmark API while use privateuse1. (#108548) 2023-09-22 14:24:18 +00:00
bottleneck [BE] Enable ruff's UP rules and autoformat utils/ (#105424) 2023-07-18 20:17:25 +00:00
data [pytorch] make IterableDataset of Iterable type (#109645) 2023-09-25 14:18:15 +00:00
hipify Revert "[ROCm] Add ROCm AMDGPU support for inductor cpp codegen (#105141)" 2023-09-12 12:29:55 +00:00
jit [BE] Enable ruff's UP rules and autoformat utils/ (#105424) 2023-07-18 20:17:25 +00:00
model_dump
tensorboard Add global_step parameter to SummaryWriter.add_hparams (#109572) 2023-09-22 12:37:01 +00:00
viz MemoryViz: print stream 0 if other streams exist (#106483) 2023-08-03 00:42:13 +00:00
__init__.py Add torch.utils to the docs page, remove dead code and fix docstrings (#105142) 2023-07-26 14:24:58 +00:00
_content_store.py Make conversions from/to sparse semi-structured always @torch.compile-d (#105272) 2023-07-18 04:51:28 +00:00
_contextlib.py grad_mode decorators without paren (#107086) 2023-08-15 05:25:33 +00:00
_cpp_extension_versioner.py
_cuda_trace.py
_device.py asarray: take the default device into consideration. (#106779) 2023-08-11 13:16:42 +00:00
_foreach_utils.py add xpu support for foreach kernels (#106021) 2023-08-09 07:51:02 +00:00
_freeze.py [BE] Enable ruff's UP rules and autoformat utils/ (#105424) 2023-07-18 20:17:25 +00:00
_mode_utils.py
_python_dispatch.py _return_and_correct_aliasing: fix for schemas with mutable tensor in kwargs (#109662) 2023-09-22 07:09:04 +00:00
_pytree.py Trace through pytree API with dynamo. (#108533) 2023-09-20 00:04:56 +00:00
_stats.py
_traceback.py Change how caching/cleanup for CapturedTraceback works (#107471) 2023-08-22 01:03:13 +00:00
_triton.py [Inductor] Generalize inductor triton backend device agnostic (#109486) 2023-09-24 07:49:20 +00:00
_zip.py
backend_registration.py Fix the api of privateuse1 in comment (#106537) 2023-08-04 16:07:26 +00:00
bundled_inputs.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
checkpoint.py Revert "Reland "Update AOTAutograd to use FunctionalTensorMode instead of C++ functionalization (#106406)" (#109906)" 2023-09-26 12:10:25 +00:00
collect_env.py [pytree] Use OpTree for PyTree manipulation (#93139) 2023-09-18 21:24:56 +00:00
cpp_backtrace.py
cpp_extension.py Fix extension rebuilding on Linux (#108613) 2023-09-06 17:58:21 +00:00
dlpack.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
file_baton.py
flop_counter.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
hooks.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
mkldnn.py add fp16 support for mkldnn conv and deconv on CPU (#99496) 2023-09-19 12:37:28 +00:00
mobile_optimizer.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
model_zoo.py
pytree.py [pytree] Use OpTree for PyTree manipulation (#93139) 2023-09-18 21:24:56 +00:00
show_pickle.py
throughput_benchmark.py [BE] Enable ruff's UP rules and autoformat utils/ (#105424) 2023-07-18 20:17:25 +00:00
weak.py [BE] Enable ruff's UP rules and autoformat utils/ (#105424) 2023-07-18 20:17:25 +00:00