Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix acl reg

This commit is contained in:
jiaorui 2023-02-16 16:05:44 +08:00
parent e6ea014082
commit 909ff33989
3 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/transform/graph_ir/op_declare/elewise_calculation_ops_declare.cc
View File

@ -659,7 +659,7 @@ INPUT_MAP(ArgMin) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(dimension)}};
ATTR_INPUT_MAP(ArgMin) = {{"axis", "dimension"}};
ATTR_MAP(ArgMin) = {{"output_type", ATTR_DESC(dtype, AnyTraits<GEType>())}};
OUTPUT_MAP(ArgMin) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(ArgMin, kArgminOpName, ADPT_DESC(ArgMin))
REG_ADPT_DESC(ArgMin, kArgMinOpName, ADPT_DESC(ArgMin))
REG_ADPT_DESC(ArgMinD, kArgMinDOpName, ADPT_DESC(ArgMin))
// Threshold

13
mindspore/ccsrc/transform/graph_ir/op_declare/image_ops_declare.cc
View File

@ -111,4 +111,17 @@ ATTR_MAP(NonMaxSuppressionWithOverlaps) = EMPTY_ATTR_MAP;
OUTPUT_MAP(NonMaxSuppressionWithOverlaps) = {{0, OUTPUT_DESC(selected_indices)}};
REG_ADPT_DESC(NonMaxSuppressionWithOverlaps, prim::kPrimNonMaxSuppressionWithOverlaps->name(),
ADPT_DESC(NonMaxSuppressionWithOverlaps))
// CombinedNonMaxSuppression
INPUT_MAP(CombinedNonMaxSuppression) = {
{1, INPUT_DESC(boxes)}, {2, INPUT_DESC(scores)}, {3, INPUT_DESC(max_output_size_per_class)},
{4, INPUT_DESC(max_total_size)}, {5, INPUT_DESC(iou_threshold)}, {6, INPUT_DESC(score_threshold)}};
ATTR_MAP(CombinedNonMaxSuppression) = {{"pad_per_class", ATTR_DESC(pad_per_class, AnyTraits<bool>())},
{"clip_boxes", ATTR_DESC(clip_boxes, AnyTraits<bool>())}};
OUTPUT_MAP(CombinedNonMaxSuppression) = {{0, OUTPUT_DESC(nmsed_boxes)},
{1, OUTPUT_DESC(nmsed_scores)},
{2, OUTPUT_DESC(nmsed_classes)},
{3, OUTPUT_DESC(valid_detections)}};
REG_ADPT_DESC(CombinedNonMaxSuppression, prim::kPrimCombinedNonMaxSuppression->name(),
ADPT_DESC(CombinedNonMaxSuppression))
} // namespace mindspore::transform

3
mindspore/ccsrc/transform/graph_ir/op_declare/image_ops_declare.h
View File

@ -51,5 +51,8 @@ DECLARE_OP_USE_OUTPUT(RGBToHSV)
DECLARE_OP_ADAPTER(NonMaxSuppressionWithOverlaps)
DECLARE_OP_USE_OUTPUT(NonMaxSuppressionWithOverlaps)
DECLARE_OP_ADAPTER(CombinedNonMaxSuppression)
DECLARE_OP_USE_OUTPUT(CombinedNonMaxSuppression)
} // namespace mindspore::transform
#endif // MINDSPORE_CCSRC_TRANSFORM_GRAPH_IR_OP_DECLARE_IMAGE_OPS_DECLARE_H_