Working GPU normalize transforms

apache · Jan 8, 2019 · 0b5dd72 · 0b5dd72
1 parent 853caa3
commit 0b5dd72
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/operator/image/normalize_op.cu b/src/operator/image/normalize_op.cu
@@ -28,8 +28,7 @@ namespace op {
 namespace image {
 
 NNVM_REGISTER_OP(_image_normalize)
-.set_attr<FCompute>("FCompute", NormalizeOpForward<gpu>)
-.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseNone{ "_copy" });
+.set_attr<FCompute>("FCompute<gpu>", NormalizeOpForward<gpu>);
 
 }  // namespace image
 }  // namespace op