saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/quantization/core
History
Salil Desai 8d7242a18b [PyTorch Edge] Add Quantized Softmax Op (Naive Implementation) (#75017) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75017

This version just does dequantize, fp32 softmax, quantize.
Another version of actual quantized softmax using qnnpack will be added next

Test Plan:
From fbcode:
```buck test caffe2/test:quantization -- test_qsoftmax```

Benchmarking: See summary of D34996486

Reviewed By: kimishpatel

Differential Revision: D34943147

fbshipit-source-id: 426a0780803597a21460139c67960891d6e9cc81
(cherry picked from commit 524eede541773299fc015f47c6cd6275ed5cf421)
2022-03-31 19:32:04 +00:00
..
__init__.py
test_quantized_functional.py [quantized] Add bilinear quantized grid_sample (#66879) 2021-11-01 14:44:26 -07:00
test_quantized_module.py Add onednn quant backend (#74137) 2022-03-15 01:28:21 +00:00
test_quantized_op.py [PyTorch Edge] Add Quantized Softmax Op (Naive Implementation) (#75017) 2022-03-31 19:32:04 +00:00
test_quantized_tensor.py [Quant][core][improvement] Enabled slicing on per-channel quantized tensors (support is limited to the a contiguous sliced tensor); added correpsonding test case (#71269) 2022-02-04 20:12:15 +00:00
test_workflow_module.py [quant] Fix qmin/qmax when using customized qrange (#74717) 2022-03-31 07:49:17 +00:00
test_workflow_ops.py Deprecates _aminmax (#71576) 2022-01-21 09:23:29 +00:00