From f707be65fd2edb97abdc2e307bff8aceddd11d09 Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Mon, 13 Apr 2020 10:37:27 +0800 Subject: [PATCH] adapt Relu6 and ROIAlign --- graphengine | 2 +- mindspore/ccsrc/transform/op_declare.cc | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/graphengine b/graphengine index 32f0696599..71e3e4ac0f 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 32f06965993ca25ea936239d284e992c5512d2ff +Subproject commit 71e3e4ac0fd9a1a229f0f07ba273162d27bdbb65 diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc index 4149280404..858b9b6b39 100644 --- a/mindspore/ccsrc/transform/op_declare.cc +++ b/mindspore/ccsrc/transform/op_declare.cc @@ -477,9 +477,9 @@ OUTPUT_MAP(ApplyAdam) = {{0, OUTPUT_DESC(var)}}; #endif // Relu6 -INPUT_MAP(Relu6) = {{1, INPUT_DESC(features)}}; +INPUT_MAP(Relu6) = {{1, INPUT_DESC(x)}}; ATTR_MAP(Relu6) = EMPTY_ATTR_MAP; -OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(activations)}}; +OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(y)}}; // Relu6Grad INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(gradients)}, {2, INPUT_DESC(features)}}; @@ -569,7 +569,7 @@ OUTPUT_MAP(GatherNd) = {{0, OUTPUT_DESC(y)}}; // ROIAlign INPUT_MAP(ROIAlign) = {{1, INPUT_DESC(features)}, {2, INPUT_DESC(rois)}}; -OUTPUT_MAP(ROIAlign) = {{0, OUTPUT_DESC(output)}}; +OUTPUT_MAP(ROIAlign) = {{0, OUTPUT_DESC(y)}}; ATTR_MAP(ROIAlign) = {{"pooled_height", ATTR_DESC(pooled_height, AnyTraits())}, {"pooled_width", ATTR_DESC(pooled_width, AnyTraits())}, {"spatial_scale", ATTR_DESC(spatial_scale, AnyTraits())},