!7 adapt relu6grad

Merge pull request !7 from zhaozhenlong/adapte-relu6grad

!7 adapt relu6grad
Merge pull request !7 from zhaozhenlong/adapte-relu6grad
3ab402e1 · mindspore-ci-bot · Gitee · 1a3c06a9 · 9862dea3 · 49351fb7
显示空白变更内容
内联并排

Showing with 3 addition and 3 deletion

graphengine graphengine +1 -1

mindspore/ccsrc/transform/op_declare.cc mindspore/ccsrc/transform/op_declare.cc +2 -2

未找到文件。
--- a/graphengine @ 49351fb7
+++ b/graphengine @ 49351fb7
-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)}};