diff --git a/src/operator/tensor/broadcast_reduce_op_value.cu b/src/operator/tensor/broadcast_reduce_op_value.cu index f1b19b4a8574..5065b9fcc7bb 100644 --- a/src/operator/tensor/broadcast_reduce_op_value.cu +++ b/src/operator/tensor/broadcast_reduce_op_value.cu @@ -104,5 +104,9 @@ NNVM_REGISTER_OP(norm) .set_attr("FCompute", L2NormCompute) .set_attr("FComputeEx", L2NormComputeEx); +NNVM_REGISTER_OP(_backward_norm) +.set_attr("FCompute", L2NormGradCompute) +.set_attr("FComputeEx", L2NormGradComputeEx); + } // namespace op } // namespace mxnet