Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/FBGEMM.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/bench
diff options
context:
space:
mode:
authorDaya Khudia <dskhudia@fb.com>2019-09-11 21:47:58 +0300
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>2019-09-11 21:52:06 +0300
commit415035019ccbca2b11b62f1503fdd61e8bc59b10 (patch)
treea970118ee111cb6531ddc7527f720fd94c5fa0cf /bench
parent685b1855d739868839cd8f774d987c1a4599b138 (diff)
API changes to take unquantized bias for depthwise conv
Summary: Changing interface for on the fly bias quantization Also adding code to quantize bias on the fly Reviewed By: jianyuh Differential Revision: D17099709 fbshipit-source-id: 5cca79189c00710e703044350260a9fcaca77bb3
Diffstat (limited to 'bench')
-rw-r--r--bench/DepthwiseBenchmark.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/bench/DepthwiseBenchmark.cc b/bench/DepthwiseBenchmark.cc
index 4f36e6c..6b2f8b8 100644
--- a/bench/DepthwiseBenchmark.cc
+++ b/bench/DepthwiseBenchmark.cc
@@ -267,6 +267,7 @@ int main() {
col_offsets.data(),
bias.data(),
false, /* fuse_relu */
+ 1.0f, /* act_scale * w_scale */
tid,
num_threads);
}