已合并
add reduction for ScatterElements #6013
lu-yukun创建于 6月12日
add reduction for ScatterElements #6013
已合并
共 1 个文件变更+10-0
| @@ -30,6 +30,16 @@ static Status ParseParamsScatterElements(const Message* op_src, ge::Operator& o | |||
| 30 | } | 30 | } |
| 31 | } | 31 | } |
| 32 | op_dest.SetAttr("axis", axis); | 32 | op_dest.SetAttr("axis", axis); |
| 33 | + | ||
| 34 | + std::string reduction = "none"; | ||
| 35 | + for (const auto& attr : node->attribute()) { | ||
| 36 | + if (attr.name() == "reduction" && attr.type() == ge::onnx::AttributeProto::STRING) { | ||
| 37 | + reduction = attr.s(); | ||
| 38 | + break; | ||
| 39 | + } | ||
| 40 | + } | ||
| 41 | + op_dest.SetAttr("reduction", reduction); | ||
| 42 | + | ||
| 33 | return SUCCESS; | 43 | return SUCCESS; |
| 34 | } | 44 | } |
| 35 | 45 | ||