diff --git a/src/operator/numpy/np_elemwise_broadcast_op_extended.cc b/src/operator/numpy/np_elemwise_broadcast_op_extended.cc index 7bb3bbae3d6b..5a69cd51bcd4 100644 --- a/src/operator/numpy/np_elemwise_broadcast_op_extended.cc +++ b/src/operator/numpy/np_elemwise_broadcast_op_extended.cc @@ -377,6 +377,7 @@ MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(broadcast_fmax) .set_attr("FGradient", ElemwiseGradUseIn{"_backward_broadcast_fmax"}); NNVM_REGISTER_OP(_backward_broadcast_fmax) +.add_alias("_backward_npi_fmax") .set_num_inputs(3) .set_num_outputs(2) .set_attr("TIsBackward", true) @@ -398,6 +399,7 @@ MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_fmax_scalar) .add_alias("_npi_fmax_scalar"); MXNET_OPERATOR_REGISTER_BINARY(_backward_fmax_scalar) +.add_alias("_backward_npi_fmax_scalar") .add_argument("scalar", "float", "scalar value") .set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); }) .set_attr("FCompute", BinaryScalarOp::Backward); @@ -408,6 +410,7 @@ MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(broadcast_fmin) .set_attr("FGradient", ElemwiseGradUseIn{"_backward_broadcast_fmin"}); NNVM_REGISTER_OP(_backward_broadcast_fmin) +.add_alias("_backward_npi_fmin") .set_num_inputs(3) .set_num_outputs(2) .set_attr("TIsBackward", true) @@ -429,6 +432,7 @@ MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_fmin_scalar) .add_alias("_npi_fmin_scalar"); MXNET_OPERATOR_REGISTER_BINARY(_backward_fmin_scalar) +.add_alias("_backward_npi_fmin_scalar") .add_argument("scalar", "float", "scalar value") .set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); }) .set_attr("FCompute", BinaryScalarOp::Backward);