[Relay] Option to select which convolution layers are quantized. (#3173)
authorJosh Fromm <jwfromm@uw.edu>
Thu, 16 May 2019 02:03:40 +0000 (19:03 -0700)
committerWuwei Lin <vincentl13x@gmail.com>
Thu, 16 May 2019 02:03:40 +0000 (10:03 +0800)
commit36702a7678573fa4ff9ea04c3920e60d590496c9
treebaea0313116585fa6826e59b43f3b92402943050
parent0f2a3086fc0c95d8fc72ae9356d48edeebd2001d
[Relay] Option to select which convolution layers are quantized. (#3173)

* Stashing for later maybe.

* Added new option to leave specific layers unquantized.

* Better error checking.

* remove unneeded import

* tab to spaces

* pylint fixes

* more pylint fixes
python/tvm/relay/quantize/_annotate.py
python/tvm/relay/quantize/quantize.py
src/relay/pass/quantize.cc
src/relay/pass/quantize.h
topi/python/topi/cuda/conv2d.py