pytorch/torch
2024-06-04 00:49:04 +00:00
..
_awaits
_C [ROCm] TunableOp improvements (#124362) 2024-06-03 22:30:11 +00:00
_C_flatbuffer
_custom_op [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
_decomp [BE]: Apply PLR1736 fixes (unnecessary index lookup) (#127716) 2024-06-03 17:22:13 +00:00
_dispatch
_dynamo Revert "Add aten._unsafe_masked_index (#116491)" 2024-06-03 15:51:50 +00:00
_export [torchbind] always fakify script object by default in non-strict export (#127116) 2024-06-03 21:38:57 +00:00
_functorch [torchbind] always fakify script object by default in non-strict export (#127116) 2024-06-03 21:38:57 +00:00
_higher_order_ops Revert "[BE] wrap deprecated function/class with typing_extensions.deprecated (#126898)" 2024-05-31 19:47:24 +00:00
_inductor [inductor] fix linear_add_bias path (#127597) 2024-06-04 00:39:01 +00:00
_lazy
_library [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
_logging Add structured logging for tensor fakeification (#126879) 2024-05-31 01:58:44 +00:00
_numpy
_prims [custom_op] use new python custom ops API on prims ops (#124665) 2024-05-22 17:48:33 +00:00
_prims_common [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
_refs Add guard_size_oblivious to vector_norm (#126772) 2024-05-21 19:53:21 +00:00
_strobelight Add compile time profiler for non fbcode targets (#126904) 2024-05-29 05:06:37 +00:00
_subclasses Avoid reference cycle in inner closure (#127711) 2024-06-02 21:28:46 +00:00
_vendor
amp generalize custom_fwd&custom_bwd to be device-agnostic (#126531) 2024-05-25 06:48:16 +00:00
ao Revert "[BE]: Update mypy to 1.10.0 (#127717)" 2024-06-03 02:52:47 +00:00
autograd [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
backends [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
compiler Beef up the allow_in_graph docs (#127117) 2024-06-02 15:00:46 +00:00
contrib
cpu [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
csrc [2/N] Change static functions in headers to inline (#127764) 2024-06-04 00:49:04 +00:00
cuda [ROCm] TunableOp improvements (#124362) 2024-06-03 22:30:11 +00:00
distributed [RFC] Introduce Checkpointable for DCP (#127540) (#127628) 2024-06-03 21:21:55 +00:00
distributions Revert "[BE]: Update mypy to 1.10.0 (#127717)" 2024-06-03 02:52:47 +00:00
export Add deferred_runtime_assertion pass after run_decompositions (#127305) 2024-05-31 05:45:28 +00:00
fft
func
futures
fx Revert "[BE]: Update mypy to 1.10.0 (#127717)" 2024-06-03 02:52:47 +00:00
jit [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
legacy
lib [Split Build] Add option to create libtorch wheel and use it to build pytorch as a separate wheel (#126328) 2024-05-29 04:33:56 +00:00
linalg Added sorting notes for eig/eigvals (#127492) 2024-05-30 18:13:22 +00:00
masked [BE] enable UFMT for torch/masked/ (#127715) 2024-06-03 22:01:49 +00:00
monitor
mps
mtia
multiprocessing [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
nested [NestedTensor] Extend coverage for unbind when ragged_idx != 1 (#127493) 2024-06-03 17:46:12 +00:00
nn Revert "[Inductor][Flex-attention] Support different sequence lengths for Query and Key/Value (#127678)" 2024-06-03 18:07:57 +00:00
onnx [BE][Ez]: Apply PYI059 - Generic always come last (#127685) 2024-06-02 13:38:58 +00:00
optim Quick Fix on #126854, deepcopy lr and other possible base_parameters (#127190) 2024-06-03 18:06:31 +00:00
package
profiler [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
quantization
signal
sparse [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
special
testing Revert "Add aten._unsafe_masked_index (#116491)" 2024-06-03 15:51:50 +00:00
utils [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
xpu
__config__.py
__future__.py
__init__.py [Split Build] Make libtorch_global_deps accessible from libtorch wheel (#127570) 2024-06-03 15:14:29 +00:00
_appdirs.py
_classes.py
_compile.py improve eager overhead of _disable_dynamo (#127325) 2024-05-31 19:30:47 +00:00
_custom_ops.py
_deploy.py
_guards.py
_jit_internal.py
_linalg_utils.py Implement svd_lowrank and pca_lowrank for complex numbers (#125580) 2024-05-30 14:45:58 +00:00
_lobpcg.py Implement svd_lowrank and pca_lowrank for complex numbers (#125580) 2024-05-30 14:45:58 +00:00
_lowrank.py Implement svd_lowrank and pca_lowrank for complex numbers (#125580) 2024-05-30 14:45:58 +00:00
_meta_registrations.py Let us find out why we wrote foreach meta regs (#127623) 2024-06-01 13:58:18 +00:00
_namedtensor_internals.py
_ops.py Revert "Refresh OpOverloadPacket if a new OpOverload gets added (#126863)" (#127366) 2024-05-29 19:26:06 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor.py Improve the clarity of the torch.Tensor.backward doc (#127201) 2024-05-28 19:25:51 +00:00
_tensor_docs.py
_tensor_str.py [small] guard_size_oblivious, skip check for meta (#127298) 2024-05-28 17:53:08 +00:00
_torch_docs.py Add usage explanation in torch.dot ducment (#125908) 2024-05-22 20:33:12 +00:00
_utils.py Revert "[BE] wrap deprecated function/class with typing_extensions.deprecated (#126898)" 2024-05-31 19:47:24 +00:00
_utils_internal.py Add compile time profiler for non fbcode targets (#126904) 2024-05-29 05:06:37 +00:00
_VF.py
_vmap_internals.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt Replace python::python with Python::Module (#127485) 2024-05-31 05:57:05 +00:00
custom_class.h
custom_class_detail.h
extension.h
functional.py
hub.py [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
library.h [Reland] Verify types in custom op schemas (#126861) 2024-05-23 19:53:52 +00:00
library.py [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
overrides.py Relax use_count constraints for swap_tensors when AccumulateGrad holds a reference (#127313) 2024-05-30 07:06:55 +00:00
py.typed
quasirandom.py Fix SobolEngine default dtype handling (#126781) 2024-05-22 01:55:48 +00:00
random.py
README.txt
return_types.py
script.h
serialization.py [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
storage.py
torch_version.py
types.py
version.py.tpl

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.