pytorch/torch
Jerry Zhang 841f7600bb [quant][graphmode] Quantization pattern for aten::linear (#33854)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33854

att

Test Plan:
python test/test_jit.py

Imported from OSS

Differential Revision: D20493031

fbshipit-source-id: bafd0a3ba5d07327d451b3915f043db33b012b53
2020-03-17 16:36:30 -07:00
..
autograd [test all] Back out "Revert D20171428: [profiler] fix chrome tracing for profiler run with cuda" 2020-03-05 09:05:56 -08:00
backends Integrate XNNPACK with custom class for packing weights. (#34047) 2020-03-14 12:51:56 -07:00
contrib
csrc [quant][graphmode] Quantization pattern for aten::linear (#33854) 2020-03-17 16:36:30 -07:00
cuda Stop using ctypes to interface with CUDA libraries. (#33678) 2020-03-11 07:22:46 -07:00
distributed [RPC] Avoid polluting Python root logger on importing "torch" module (#34871) 2020-03-17 16:07:06 -07:00
distributions Continuous bernoulli distribution (take 2) (#34619) 2020-03-12 11:53:18 -07:00
for_onnx
jit [jit] copy unused/ignored methods to ScriptModule during compilation (#33981) 2020-03-16 10:38:59 -07:00
legacy
lib Remove all uses of AT_CHECK and replace them with TORCH_CHECK (#34846) 2020-03-17 08:59:02 -07:00
multiprocessing
nn adds quantized implementation of hard sigmoid (#34607) 2020-03-17 16:01:39 -07:00
onnx [ONNX] Fix for expand -1 dim value (#34069) 2020-03-16 15:30:20 -07:00
optim Fix typo in documentation (#34581) 2020-03-11 13:57:10 -07:00
quantization Add the quantized batch_norm3d and also batch_norm3d fused with relu operators (#34702) 2020-03-13 20:30:28 -07:00
sparse
testing fix tests that could have racy script module instantiation (#34792) 2020-03-16 23:14:07 -07:00
utils Ensure torch_cuda is linked against on Windows (#34288) 2020-03-12 12:16:44 -07:00
__config__.py
__future__.py
__init__.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
__init__.pyi.in Support for Tensor Shape Type Hint (#34595) 2020-03-13 15:16:24 -07:00
_classes.py
_jit_internal.py [JIT] Preserve qualified names on traced modules (#34395) 2020-03-09 19:23:53 -07:00
_linalg_utils.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
_lobpcg.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
_lowrank.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
_namedtensor_internals.py
_ops.py
_overrides.py Add types argument to __torch_function__ (#34303) 2020-03-17 13:32:00 -07:00
_six.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
_storage_docs.py
_tensor_docs.py
_tensor_str.py
_torch_docs.py Adds true_divide function, analogous to Python 's, JAX's, NumPy's (true) division (#34236) 2020-03-09 21:06:33 -07:00
_utils.py
_utils_internal.py
_VF.py
abi-check.cpp
CMakeLists.txt Don't run user function until all UserRRefs in the args are confirmed (#34497) 2020-03-16 18:30:06 -07:00
custom_class.h Move torchbind out of jit namespace (#34745) 2020-03-13 23:03:14 -07:00
custom_class_detail.h Move torchbind out of jit namespace (#34745) 2020-03-13 23:03:14 -07:00
extension.h
functional.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
hub.py
py.typed
quasirandom.py
random.py
README.txt
script.h
serialization.py PCA and SVD for low-rank matrices, LOBPCG for positive-defined generalized eigenvalue problem - copy (#34721) 2020-03-16 14:13:30 -07:00
storage.py
tensor.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.