pytorch/torch/utils
Kevin Tse 39fb855d91 [DataLoader] Implementing communication processes for Map-style DataPipes (#68549)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68549

cc SsnL VitalyFedyunin ejguan NivekT

Test Plan: Imported from OSS

Reviewed By: zou3519

Differential Revision: D32922676

Pulled By: NivekT

fbshipit-source-id: fd918a342214d617a489ac5acffff15b55e9b255
2021-12-08 07:27:01 -08:00
..
backcompat
benchmark [NOOP][clangformat][codemod] Enable CLANGFORMAT for some folders in caffe2/* (#67746) 2021-11-03 12:23:14 -07:00
bottleneck [profiler][refactor] Refactor the usage of legacy profiler implementation (#61931) 2021-08-03 18:51:29 -07:00
data [DataLoader] Implementing communication processes for Map-style DataPipes (#68549) 2021-12-08 07:27:01 -08:00
ffi
hipify Sparse CSR CUDA: Add block torch.addmm when mat1 is sparse (#68707) 2021-11-29 08:58:49 -08:00
model_dump model_dump: Fix memory computation when both constants and data tensors are present (#66006) 2021-10-01 16:31:06 -07:00
tensorboard Fix attr_to_scope's key in torch/utils/tensorboard/_pytorch_graph.py (#65692) 2021-10-20 14:35:29 -07:00
__init__.py
_cpp_extension_versioner.py
_crash_handler.py Use CMake for breakpad (#63186) 2021-08-19 10:42:01 -07:00
_freeze.py don't leak build time path name to runtime for frozen python modules (#65715) 2021-09-28 19:25:51 -07:00
_python_dispatch.py [Reland] Add python mode (#64360) 2021-09-16 09:02:30 -07:00
_pytree.py Teach pytrees about namedtuple (#62292) 2021-07-28 06:27:44 -07:00
_zip.py Add custom zipper script to zip python modules for torch.deploy (#67006) 2021-11-06 11:49:02 -07:00
bundled_inputs.py [SmartCompose][OnDevice]fix function name bug in mobile export & Script to convert mobile model (#66915) 2021-10-20 18:14:51 -07:00
checkpoint.py [Reland][Autograd/Checkpoint] Checkpoint implementation without reentrant autograd (#69508) 2021-12-07 16:31:23 -08:00
collect_env.py Improve pip package determination (#63321) 2021-08-16 13:54:39 -07:00
cpp_extension.py Add support for C++ frontend wrapper on Linux (#69094) 2021-12-02 16:47:00 -08:00
dlpack.py Fix DLPack CUDA stream convention (#67618) 2021-11-18 08:36:05 -08:00
file_baton.py
hooks.py Fix full backward hook when grad is disabled (#65335) 2021-09-20 13:31:19 -07:00
mkldnn.py
mobile_optimizer.py
model_zoo.py
show_pickle.py
throughput_benchmark.py