[incubator-mxnet] branch master updated: fix gradient number and document for box_iou operator (#10823)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhreshold 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 33c58dc  fix gradient number and document for box_iou operator (#10823)
33c58dc is described below

commit 33c58dc95e7fefbe89c52076261935496a79a02b
Author: JackieWu <wk...@live.cn>
AuthorDate: Wed May 9 08:54:58 2018 +0800

    fix gradient number and document for box_iou operator (#10823)
---
 src/operator/contrib/bounding_box.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/operator/contrib/bounding_box.cc b/src/operator/contrib/bounding_box.cc
index 288fe44..53052ad 100644
--- a/src/operator/contrib/bounding_box.cc
+++ b/src/operator/contrib/bounding_box.cc
@@ -117,7 +117,7 @@ NNVM_REGISTER_OP(_contrib_box_iou)
   Example::
 
     x = [[0.5, 0.5, 1.0, 1.0], [0.0, 0.0, 0.5, 0.5]]
-    y = [0.25, 0.25, 0.75, 0.75]
+    y = [[0.25, 0.25, 0.75, 0.75]]
     box_iou(x, y, format='corner') = [[0.1428], [0.1428]]
 
 )doc" ADD_FILELINE)
@@ -137,8 +137,8 @@ NNVM_REGISTER_OP(_contrib_box_iou)
 .add_arguments(BoxOverlapParam::__FIELDS__());
 
 NNVM_REGISTER_OP(_backward_contrib_box_iou)
-.set_num_inputs(2)
-.set_num_outputs(1)
+.set_num_inputs(1)
+.set_num_outputs(2)
 .set_attr_parser(ParamParser<BoxOverlapParam>)
 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
 .set_attr<FCompute>("FCompute<cpu>", BoxOverlapBackward<cpu>)

-- 
To stop receiving notification emails like this one, please contact
zhreshold@apache.org.