set inputs to 3 for lrn

apache · Aug 9, 2018 · f873297 · f873297
1 parent db6f1cd
commit f873297
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/operator/nn/lrn.cc b/src/operator/nn/lrn.cc
@@ -198,6 +198,7 @@ number of kernels in the layer.
 .add_arguments(LRNParam::__FIELDS__());
 
 NNVM_REGISTER_OP(_backward_LRN)
+.set_num_inputs(3)
 .set_num_outputs(1)
 .set_attr_parser(ParamParser<LRNParam>)
 .set_attr<FInferStorageType>("FInferStorageType", LRNBackwardInferStorageType)