Browse Source

!7 adapt relu6grad

Merge pull request !7 from zhaozhenlong/adapte-relu6grad
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
3ab402e110
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      graphengine
  2. +2
    -2
      mindspore/ccsrc/transform/op_declare.cc

+ 1
- 1
graphengine

@@ -1 +1 @@
Subproject commit 5f763679fa33de1608d07f7651c6f16012b953ea
Subproject commit 49351fb73ac7786b9ed9e807568a1a8e06183693

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

@@ -506,9 +506,9 @@ ATTR_MAP(Relu6) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(activations)}}; OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(activations)}};


// Relu6Grad // 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; ATTR_MAP(Relu6Grad) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(z)}};
OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(backprops)}};


// ResizeBilinearGrad // ResizeBilinearGrad
INPUT_MAP(ResizeBilinearGrad) = {{1, INPUT_DESC(grads)}, {2, INPUT_DESC(original_image)}}; INPUT_MAP(ResizeBilinearGrad) = {{1, INPUT_DESC(grads)}, {2, INPUT_DESC(original_image)}};


Loading…
Cancel
Save