projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: fbd690c) | patch
bug fix when we prepack weight and bias together (#17145)
authorJongsoo Park <jongsoo@fb.com>
Fri, 15 Feb 2019 17:17:53 +0000 (09:17 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 15 Feb 2019 17:21:20 +0000 (09:21 -0800)
commit90fc6133b2fed145536f3b479adf0e6f13fd9a6c
tree9821431bff797af446e937af33307636eb5c3fc2tree | snapshot
parentfbd690c1fec0651ee9e6cc07ddcb12217ffb31bccommit | diff
bug fix when we prepack weight and bias together (#17145)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17145

Prepacked weight contains both weight and bias, so the bias should be obtained from input index 1, not from 2

Reviewed By: jianyuh

Differential Revision: D14097281

fbshipit-source-id: b8b836b85a7b240e2fd1734377c46d9bf2ce3390
caffe2/quantization/server/conv_dnnlowp_op.cc diff | blob | history
caffe2/quantization/server/fully_connected_dnnlowp_op.cc diff | blob | history
Domain: Machine Learning / ML Framework;