From 9862dea3cffea33119987882471db9cf35dda458 Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Mon, 30 Mar 2020 09:34:34 +0800 Subject: [PATCH] adapt relu6grad and graphengine modified --- graphengine | 2 +- mindspore/ccsrc/transform/op_declare.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/graphengine b/graphengine index 5f763679fa..49351fb73a 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 78b949c525..82dcd62572 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)}};