Browse Source

adapt Relu6 and ROIAlign

tags/v0.3.0-alpha
zhaozhenlong 5 years ago
parent
commit
f707be65fd
2 changed files with 4 additions and 4 deletions
  1. +1
    -1
      graphengine
  2. +3
    -3
      mindspore/ccsrc/transform/op_declare.cc

+ 1
- 1
graphengine

@@ -1 +1 @@
Subproject commit 32f06965993ca25ea936239d284e992c5512d2ff
Subproject commit 71e3e4ac0fd9a1a229f0f07ba273162d27bdbb65

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

@@ -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<int>())},
{"pooled_width", ATTR_DESC(pooled_width, AnyTraits<int>())},
{"spatial_scale", ATTR_DESC(spatial_scale, AnyTraits<float>())},


Loading…
Cancel
Save