pytorch/torch
Victor Quach ee5a97de11 Register Saved Tensors hooks (#60663)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60663

Test Plan: Imported from OSS

Reviewed By: soulitzer

Differential Revision: D29466223

fbshipit-source-id: 65dc3a935c18a0e6b93a37e24543c696e6ae0321
2021-07-15 08:09:55 -07:00
..
_C Revert FC workaround for ReflectionPad3d (#61308) 2021-07-12 14:19:07 -07:00
ao [pruner] fix activation handles logic (#61592) 2021-07-14 11:07:23 -07:00
autograd Add neg bit (#56058) 2021-07-13 13:50:42 -07:00
backends Add option to specify custom NNAPI serializer (#61025) 2021-07-09 15:27:10 -07:00
contrib
cpu
csrc Register Saved Tensors hooks (#60663) 2021-07-15 08:09:55 -07:00
cuda
distributed Remove _broadcast_object() from ZeroRedundancyOptimizer (#61539) 2021-07-14 17:36:30 -07:00
distributions Improve error message on invalid values to Distribution methods (#61056) 2021-07-06 15:44:55 -07:00
fft
for_onnx
futures fix torch.futures docstring examples (#61029) 2021-07-07 11:47:55 -07:00
fx add AdaptiveAvgPooling2D (#61239) 2021-07-10 20:07:14 -07:00
jit
legacy
lib
linalg warn about SVD outputs not supporting backprop (#61037) 2021-07-12 12:55:37 -07:00
multiprocessing
nn Refactor non-joined process computation (#61555) 2021-07-14 08:20:40 -07:00
onnx [ONNX] Update documentation (#58712) (#60249) 2021-07-08 16:29:32 -07:00
optim Fix CosineAnnealingWarmRestart annotation (#61106) 2021-07-09 08:28:18 -07:00
package Revert D29639797: [package] error if we try to mock a module in 3.6 2021-07-09 19:31:04 -07:00
profiler
quantization [quant] Input-Weight Equalization - ConvReLU support (#61350) 2021-07-13 14:00:40 -07:00
sparse
special [special] migrate log_softmax (#60512) 2021-07-12 11:01:25 -07:00
testing Remove _broadcast_object() from ZeroRedundancyOptimizer (#61539) 2021-07-14 17:36:30 -07:00
utils Update cpp_extension.py (#61484) 2021-07-13 07:11:58 -07:00
__config__.py
__future__.py
__init__.py
_appdirs.py
_autograd_functions.py
_classes.py
_deploy.py
_jit_internal.py Support RRefs that contain torch.cuda.Event (#61354) 2021-07-08 15:33:08 -07:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_namedtensor_internals.py
_ops.py
_python_dispatcher.py
_six.py
_storage_docs.py
_tensor.py Add neg bit (#56058) 2021-07-13 13:50:42 -07:00
_tensor_docs.py Add neg bit (#56058) 2021-07-13 13:50:42 -07:00
_tensor_str.py Add neg bit (#56058) 2021-07-13 13:50:42 -07:00
_torch_docs.py Add neg bit (#56058) 2021-07-13 13:50:42 -07:00
_utils.py torch._utils.ExceptionWrapper: fix for Exceptions with multiple args (#58131) 2021-07-12 09:04:36 -07:00
_utils_internal.py
_VF.py
_vmap_internals.py
abi-check.cpp
CMakeLists.txt
custom_class.h
custom_class_detail.h
deploy.h
extension.h
functional.py Allow dims=0 in torch.tensordot call (#61331) 2021-07-10 17:05:20 -07:00
hub.py
library.h
overrides.py Add neg bit (#56058) 2021-07-13 13:50:42 -07:00
py.typed
quasirandom.py
random.py
README.txt
script.h
serialization.py
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.