You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by ha...@apache.org on 2018/05/07 23:26:45 UTC
[incubator-mxnet] branch master updated: Fix quantized pooling doc
(#10841)
This is an automated email from the ASF dual-hosted git repository.
haibin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 6cfc614 Fix quantized pooling doc (#10841)
6cfc614 is described below
commit 6cfc614e75a2ebc2a28b82b04adb1f57c7216e61
Author: reminisce <wu...@gmail.com>
AuthorDate: Mon May 7 16:26:40 2018 -0700
Fix quantized pooling doc (#10841)
---
src/operator/quantization/quantized_pooling.cc | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/src/operator/quantization/quantized_pooling.cc b/src/operator/quantization/quantized_pooling.cc
index 71f4e73..a3105eb 100644
--- a/src/operator/quantization/quantized_pooling.cc
+++ b/src/operator/quantization/quantized_pooling.cc
@@ -92,6 +92,13 @@ bool QuantizedPoolingType(const nnvm::NodeAttrs& attrs,
}
NNVM_REGISTER_OP(_contrib_quantized_pooling)
+.describe(R"code(Pooling operator for input and output data type of int8.
+The input and output data comes with min and max thresholds for quantizing
+the float32 data into int8.
+
+.. Note::
+ This operator only supports forward propogation. DO NOT use it in training.
+ This operator only supports `pool_type` of `avg` or `max`.)code" ADD_FILELINE)
.set_num_inputs(3)
.set_num_outputs(3)
.set_attr_parser(ParamParser<PoolingParam>)
@@ -118,13 +125,6 @@ NNVM_REGISTER_OP(_contrib_quantized_pooling)
.add_arguments(PoolingParam::__FIELDS__());
NNVM_REGISTER_OP(Pooling)
-.describe(R"code(Pooling operator for input and output data type of int8.
-The input and output data comes with min and max thresholds for quantizing
-the float32 data into int8.
-
-.. Note::
- This operator only supports forward propogation. DO NOT use it in training.
- This operator only supports `pool_type` of `avg` or `max`.)code" ADD_FILELINE)
.set_attr<FQuantizedOp>("FQuantizedOp", [](const NodeAttrs& attrs) {
PoolingParam param;
param.Init(attrs.dict);
--
To stop receiving notification emails like this one, please contact
haibin@apache.org.