[incubator-mxnet] branch master updated: support L2Norm backward for GPU. (#10874)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhasheng 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 8b53a3d  support L2Norm backward for GPU. (#10874)
8b53a3d is described below

commit 8b53a3d2311154201cb4ade5c8f42f5c226eefb1
Author: Da Zheng <zh...@gmail.com>
AuthorDate: Sat May 12 23:19:37 2018 -0700

    support L2Norm backward for GPU. (#10874)
    
    * support gpu.
    
    * Fix.
---
 src/operator/tensor/broadcast_reduce_op_value.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/operator/tensor/broadcast_reduce_op_value.cu b/src/operator/tensor/broadcast_reduce_op_value.cu
index f1b19b4..5065b9f 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>("FCompute<gpu>", L2NormCompute<gpu>)
 .set_attr<FComputeEx>("FComputeEx<gpu>", L2NormComputeEx<gpu>);
 
+NNVM_REGISTER_OP(_backward_norm)
+.set_attr<FCompute>("FCompute<gpu>", L2NormGradCompute<gpu>)
+.set_attr<FComputeEx>("FComputeEx<gpu>", L2NormGradComputeEx<gpu>);
+
 }  // namespace op
 }  // namespace mxnet

-- 
To stop receiving notification emails like this one, please contact
zhasheng@apache.org.