diff --git a/graphengine b/graphengine index 5f763679fa33de1608d07f7651c6f16012b953ea..49351fb73ac7786b9ed9e807568a1a8e06183693 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 5f763679fa33de1608d07f7651c6f16012b953ea +Subproject commit 49351fb73ac7786b9ed9e807568a1a8e06183693 diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc index 78b949c52558791e691661780e15b3fe5d07c842..82dcd625720ca8661e377bd0ba5783cc4469bf0a 100755 --- 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)}};