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 <w...@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 zhresh...@apache.org.