!7 adapt relu6grad

Merge pull request !7 from zhaozhenlong/adapte-relu6grad
5 years ago · 3ab402e110
parent 1a3c06a948 9862dea3cf
commit 3ab402e110
2 changed files with 3 additions and 3 deletions
--- a/2
+++ b/2
@ -1 +1 @@
-Subproject commit 5f763679fa33de1608d07f7651c6f16012b953ea
+Subproject commit 49351fb73ac7786b9ed9e807568a1a8e06183693
--- a/mindspore/ccsrc/transform/op_declare.cc
+++ b/mindspore/ccsrc/transform/op_declare.cc
@ -506,9 +506,9 @@ ATTR_MAP(Relu6) = EMPTY_ATTR_MAP;
 OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(activations)}};
 // Relu6Grad
-INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(dy)}, {2, INPUT_DESC(y)}};
+INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(gradients)}, {2, INPUT_DESC(features)}};
 ATTR_MAP(Relu6Grad) = EMPTY_ATTR_MAP;
-OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(z)}};
+OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(backprops)}};
 // ResizeBilinearGrad
 INPUT_MAP(ResizeBilinearGrad) = {{1, INPUT_DESC(grads)}, {2, INPUT_DESC(original_image)}};
		`@ -1 +1 @@`
			`Subproject commit 5f763679fa33de1608d07f7651c6f16012b953ea`				`Subproject commit 49351fb73ac7786b9ed9e807568a1a8e06183693`