pytorch/torch
2022-09-09 15:19:33 +00:00
..
_C Add underlying_store property for PrefixStore (#84640) 2022-09-07 23:02:47 +00:00
_C_flatbuffer
_decomp Added mv, mm, and binary_cross_entropy_with_logits decomps (#84451) 2022-09-08 17:56:18 +00:00
_dispatch
_lazy Add step closures (#84300) 2022-09-06 20:55:34 +00:00
_masked [maskedtensor] adding reductions (#82839) 2022-09-06 15:01:35 +00:00
_prims Add config to PrimTorch's nvFuser executor (#84482) 2022-09-09 07:58:21 +00:00
_prims_common Add SymFloat, support SymInt to SymFloat conversion (#84284) 2022-09-03 01:30:32 +00:00
_refs [primTorch] Add refs for reshape_as, view_as, unify tests (#84222) 2022-09-01 16:14:34 +00:00
_subclasses Redo how custom/python_custom methods on TensorImpl work (#84641) 2022-09-09 13:41:13 +00:00
amp
ao [quant] Subpackage import in nn.quantized (#84141) 2022-09-01 11:35:03 +00:00
autograd
backends
contrib
cpu
csrc Add synchronize hooks (#84427) 2022-09-09 13:56:59 +00:00
cuda Add CSAN support for CPU synchronizations (#84428) 2022-09-09 15:19:33 +00:00
distributed [FSDP] Ensure that all ranks use the same order to iterate through optimizer states (#84654) 2022-09-09 07:19:01 +00:00
distributions remove data-dependent shapes from some distributions (#84322) 2022-08-31 09:37:55 +00:00
fft
futures
fx [torch.fx.wrap] Use callable / function.__name__ instead of function.__code__.co_name (#84373) 2022-09-09 05:44:29 +00:00
jit
legacy
lib
linalg
masked [maskedtensor] adding reductions (#82839) 2022-09-06 15:01:35 +00:00
monitor
multiprocessing
nested
nn Disable Transformer/MHA fast path when autocast is enabled (#84722) 2022-09-09 01:15:24 +00:00
onnx [ONNX] Fix node attributes when namespace is aten (#84211) 2022-09-09 01:52:39 +00:00
optim fix: Allowed optimizers with more than 2 betas (#84486) 2022-09-06 19:24:10 +00:00
package fix typo in torch/package/_mock.py (#84508) 2022-09-05 16:48:34 +00:00
profiler
quantization [Pytorch][quantization][ondevice] Add a wrapper API for server side prep (#83742) 2022-08-29 17:55:26 +00:00
sparse
special
testing Fix linalg.norm sample inputs function and related failures (#84452) 2022-09-08 17:07:58 +00:00
utils [DataPipe] Fix mishandling of exception message when error is not iterable (#84676) 2022-09-09 14:34:13 +00:00
__config__.py
__future__.py
__init__.py Add CUDA Sanitizer (#83984) 2022-09-07 16:55:03 +00:00
_appdirs.py
_classes.py
_deploy.py
_jit_internal.py
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py remove unneeded _to_copy meta (#84460) 2022-09-02 18:08:39 +00:00
_namedtensor_internals.py
_ops.py
_python_dispatcher.py
_six.py
_sources.py
_storage_docs.py
_tensor.py Fix split torch_function handling (#83866) 2022-08-30 18:03:32 +00:00
_tensor_docs.py
_tensor_str.py
_torch_docs.py Sparse Compressed Transpose add support for Batch dims and BSR/BSC layouts (#82122) 2022-09-02 17:42:58 +00:00
_utils.py
_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 Revert "Dispatch torch.norm to linalg.vector_norm and linalg.matrix_norm (#81761)" 2022-09-06 22:31:14 +00:00
hub.py
library.h
library.py
overrides.py [Modes] remove inner constructor kwarg (#83925) 2022-08-31 00:05:56 +00:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py
storage.py
torch_version.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.