Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

add infer_type for regression ops, fix issues#9847 #9848

Merged
piiswrong merged 2 commits into from
Feb 27, 2018
Merged

add infer_type for regression ops, fix issues#9847 #9848

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
00ae900
add infer_type for regression ops
ZiyueHuang Feb 21, 2018
02a365c
trigger CI
ZiyueHuang Feb 23, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/operator/regression_output.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
*/

#include "./regression_output-inl.h"
#include "./elemwise_op_common.h"


#define MXNET_OPERATOR_REGISTER_REGRESSION_FWD(__name$, __kernel$, __bwdop$) \
NNVM_REGISTER_OP(__name$) \
Expand All @@ -33,6 +35,7 @@
return std::vector<std::string>{"data", "label"}; \
}) \
.set_attr<nnvm::FInferShape>("FInferShape", RegressionOpShape) \
.set_attr<nnvm::FInferType>("FInferType", ElemwiseType<2, 1>) \
.set_attr<nnvm::FGradient>("FGradient", RegressionOpGrad{__bwdop$}) \
.set_attr<nnvm::FInplaceOption>("FInplaceOption", \
[](const NodeAttrs& attrs){ \
Expand All @@ -48,6 +51,7 @@
.set_num_inputs(2) \
.set_num_outputs(2) \
.set_attr_parser(ParamParser<RegressionOutputParam>) \
.set_attr<nnvm::FInferType>("FInferType", ElemwiseType<2, 2>) \
.set_attr<nnvm::TIsBackward>("TIsBackward", true) \
.set_attr<nnvm::FInplaceOption>("FInplaceOption", \
[](const NodeAttrs& attrs){ \
Expand Down