pytorch/torch
2021-06-30 16:10:31 +00:00
..
_C Expose raw saved tensors for custom functions (#60551) 2021-06-29 17:21:52 -07:00
ao [pruning] Speedup activation reconstruction (#60683) 2021-06-28 12:58:21 -07:00
autograd Expose raw saved tensors for custom functions (#60551) 2021-06-29 17:21:52 -07:00
backends NNAPI: flex size support for upsample_nearest2d op (#57563) 2021-05-05 13:54:43 -07:00
contrib
cpu enable torch.cpu.amp.autocast (#57386) 2021-05-20 17:48:36 -07:00
csrc Add noop register hook (#60685) 2021-06-30 07:46:34 -07:00
cuda Don't split oversize cached blocks (#44742) 2021-06-21 11:46:08 -07:00
distributed [Model Averaging] Enforce a synchronization before allreduce parameters (#60891) 2021-06-29 01:39:40 -07:00
distributions [docs] Fix backticks in docs (#60474) 2021-06-24 06:27:41 -07:00
fft Fix FFT documentation examples and run doctests in the test suite (#60304) 2021-06-21 20:47:25 -07:00
for_onnx
futures Update docs to mention CUDA support for Future (#50048) 2021-05-11 08:26:33 -07:00
fx [FX] Add requires_grad to TensorMetadata (#60972) 2021-06-29 23:07:27 -07:00
jit don't return in __init__ functions (#60830) 2021-06-28 14:56:13 -07:00
legacy
lib Migrate THAllocator to ATen (#60325) 2021-06-24 19:42:14 -07:00
linalg [docs] Fix backticks in docs (#60474) 2021-06-24 06:27:41 -07:00
multiprocessing Fix typo in warning for spawn method (#57927) 2021-05-10 13:12:38 -07:00
nn [DDP] Fix case where new tensors with no grad_fn are returned in DDP forward. (#60882) 2021-06-29 12:50:48 -07:00
onnx List striding with arbitrary step size (#58537) 2021-06-22 11:25:23 -07:00
optim To refactor multi tensor RMSprop to functional API 2021-06-30 16:10:31 +00:00
package [package] track storages across lifetime of PackageExporter (#59735) 2021-06-29 14:16:54 -07:00
profiler [profile] Clarify record_shapes=True docstring (#59469) 2021-06-04 12:01:35 -07:00
quantization ns for fx: initial support for int8 shadows fp32 (#60419) 2021-06-30 08:08:46 -07:00
sparse
special [special] add zeta (#59623) 2021-06-24 00:00:12 -07:00
testing resolve conjugate bit in torch.testing.assert_close (#60522) 2021-06-30 01:31:30 -07:00
utils Add TorchVitals for DataLoader (#60959) 2021-06-29 14:08:32 -07:00
__config__.py
__future__.py
__init__.py Add device runtime API for the plug-in to register platform python module into torch (#59857) 2021-06-23 07:54:45 -07:00
_appdirs.py Disallow versionless Python shebangs (#58275) 2021-05-14 08:26:02 -07:00
_autograd_functions.py
_classes.py
_deploy.py
_jit_internal.py Support default values on NamedTuple fields (#54682) 2021-06-26 15:18:21 -07:00
_linalg_utils.py Deprecate symeig (#57732) 2021-05-12 02:21:35 -07:00
_lobpcg.py Remove Optional[None] annotations (#60704) 2021-06-25 15:53:58 -07:00
_lowrank.py Deprecate in docs torch.svd and change svd -> linalg_svd (#57981) 2021-05-11 18:04:10 -07:00
_namedtensor_internals.py
_ops.py [deploy] Make load_library a no-op inside a package (#58933) 2021-06-02 17:01:31 -07:00
_python_dispatcher.py
_six.py
_storage_docs.py
_tensor_docs.py Implemented torch.cov (#58311) 2021-06-29 14:02:39 -07:00
_tensor_str.py Conjugate View (#54987) 2021-06-04 14:12:41 -07:00
_tensor.py supports non-leaf inputs for autograd.backward() function (#60521) 2021-06-25 18:57:26 -07:00
_torch_docs.py Implemented torch.cov (#58311) 2021-06-29 14:02:39 -07:00
_utils_internal.py Strictly type everything in .github and tools (#59117) 2021-06-07 14:49:36 -07:00
_utils.py move flatten_dense_tensors and unflatten_dense_tensors to Native (#58006) 2021-05-12 18:18:34 -07:00
_VF.py
_vmap_internals.py Add note about improved vmap prototype to vmap docstring (#57677) 2021-05-06 06:47:18 -07:00
abi-check.cpp
CMakeLists.txt [reland] Move RPC agents to libtorch (#60170) 2021-06-18 05:15:09 -07:00
custom_class_detail.h irange for PyTorch sans jit (#59481) 2021-06-09 14:46:11 -07:00
custom_class.h don't use moved field in the second lambda (#59914) 2021-06-16 17:22:15 -07:00
deploy.h
extension.h
functional.py [docs] Fix backticks in docs (#60474) 2021-06-24 06:27:41 -07:00
hub.py
library.h HABANA Device registration key and Autograd key addition (#57094) 2021-05-12 13:07:33 -07:00
overrides.py Implemented torch.cov (#58311) 2021-06-29 14:02:39 -07:00
py.typed
quasirandom.py
random.py Improve documentation for torch.set_rng_state (#60422) 2021-06-22 07:10:50 -07:00
README.txt
script.h
serialization.py Ensure torch.save() deterministic output (#57536) 2021-05-10 11:51:55 -07:00
storage.py
types.py

Note [TH abstraction violation]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

TH/THC provide some hpp headers, which are proper C++ headers rather than
C headers.  These headers serve double duty as *internal implementation
detail* headers, whose contents should largely not be used by external
clients.

Ideally, we would not install these headers at all; instead, you should
use public functions (in headers like `THTensor.h`, NOT `THTensor.hpp`)
to manipulate these structs.  However, there are a few places
in torch/csrc where we violate this abstraction.  They are marked with
a pointer to this note.  Each of those sites will have to be refactored
when we refactor the guts of THTensor and related structures.