forked from mindspore-Ecosystem/mindspore
!7 adapt relu6grad
Merge pull request !7 from zhaozhenlong/adapte-relu6grad
This commit is contained in:
commit
3ab402e110
|
@ -1 +1 @@
|
|||
Subproject commit 5f763679fa33de1608d07f7651c6f16012b953ea
|
||||
Subproject commit 49351fb73ac7786b9ed9e807568a1a8e06183693
|
|
@ -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)}};
|
||||
|
|
Loading…
Reference in New Issue