p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

adapt relu6grad and graphengine modified

This commit is contained in:
zhaozhenlong 2020-03-30 09:34:34 +08:00 committed by chang zherui
parent 02ebf03c6e
commit 04e4f6d927
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphengine

@ -1 +1 @@
Subproject commit 092c7a1f6548cac7d40e677af3498c3c49ea2bfd
Subproject commit 78a40081fc1462ab9e2083d37e5fbe608a57dd06

2
mindspore/ccsrc/transform/op_declare.cc
View File

@ -473,7 +473,7 @@ ATTR_MAP(Relu6) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(activations)}};
// Relu6Grad
INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(features)}, {2, INPUT_DESC(gradients)}};
INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(gradients)}, {2, INPUT_DESC(features)}};
ATTR_MAP(Relu6Grad) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(backprops)}};