Browse Source

adapt relu6grad and graphengine modified

tags/v0.3.0-alpha
zhaozhenlong chang zherui 6 years ago
parent
commit
04e4f6d927
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      graphengine
  2. +1
    -1
      mindspore/ccsrc/transform/op_declare.cc

+ 1
- 1
graphengine

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

+ 1
- 1
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)}};



Loading…
Cancel
Save