已合并
unpack支持complex32类型 #3413
hw-zhangpanpan创建于 6月17日
unpack支持complex32类型 #3413
已合并
从已删除 :master合入到cann/ops-mathmaster
共 2 个文件变更+51-6
| @@ -180,6 +180,51 @@ | |||
| 180 | "value": null | 180 | "value": null |
| 181 | } | 181 | } |
| 182 | ] | 182 | ] |
| 183 | + }, | ||
| 184 | + { | ||
| 185 | + "bin_filename": "Unpack_complex32", | ||
| 186 | + "inputs": [ | ||
| 187 | + { | ||
| 188 | + "name": "x", | ||
| 189 | + "index": 0, | ||
| 190 | + "dtype": "complex32", | ||
| 191 | + "format": "ND", | ||
| 192 | + "paramType": "required", | ||
| 193 | + "shape": [ | ||
| 194 | + -2 | ||
| 195 | + ], | ||
| 196 | + "format_match_mode": "FormatDefault", | ||
| 197 | + "dtype_match_mode": "DtypeByte" | ||
| 198 | + } | ||
| 199 | + ], | ||
| 200 | + "outputs": [ | ||
| 201 | + [ | ||
| 202 | + { | ||
| 203 | + "name": "y", | ||
| 204 | + "index": 0, | ||
| 205 | + "dtype": "complex32", | ||
| 206 | + "format": "ND", | ||
| 207 | + "paramType": "dynamic", | ||
| 208 | + "shape": [ | ||
| 209 | + -2 | ||
| 210 | + ], | ||
| 211 | + "format_match_mode": "FormatDefault", | ||
| 212 | + "dtype_match_mode": "DtypeByte" | ||
| 213 | + } | ||
| 214 | + ] | ||
| 215 | + ], | ||
| 216 | + "attrs": [ | ||
| 217 | + { | ||
| 218 | + "name": "num", | ||
| 219 | + "dtype": "int", | ||
| 220 | + "value": null | ||
| 221 | + }, | ||
| 222 | + { | ||
| 223 | + "name": "axis", | ||
| 224 | + "dtype": "int", | ||
| 225 | + "value": null | ||
| 226 | + } | ||
| 227 | + ] | ||
| 183 | } | 228 | } |
| 184 | ] | 229 | ] |
| 185 | } | 230 | } |
| @@ -19,19 +19,19 @@ class Unpack : public OpDef { | |||
| 19 | this->Input("x") | 19 | this->Input("x") |
| 20 | .ParamType(REQUIRED) | 20 | .ParamType(REQUIRED) |
| 21 | .DataType({ge::DT_INT64, ge::DT_UINT64, ge::DT_INT32, ge::DT_UINT32, ge::DT_FLOAT, ge::DT_FLOAT16, | 21 | .DataType({ge::DT_INT64, ge::DT_UINT64, ge::DT_INT32, ge::DT_UINT32, ge::DT_FLOAT, ge::DT_FLOAT16, |
| 22 | - ge::DT_INT8, ge::DT_UINT8, ge::DT_BF16, ge::DT_INT16, ge::DT_UINT16, ge::DT_BOOL}) | 22 | + ge::DT_INT8, ge::DT_UINT8, ge::DT_BF16, ge::DT_INT16, ge::DT_UINT16, ge::DT_BOOL, ge::DT_COMPLEX32}) |
| 23 | .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 23 | .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, |
| 24 | - ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) | 24 | + ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 25 | .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 25 | .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, |
| 26 | - ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); | 26 | + ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); |
| 27 | this->Output("y") | 27 | this->Output("y") |
| 28 | .ParamType(DYNAMIC) | 28 | .ParamType(DYNAMIC) |
| 29 | .DataType({ge::DT_INT64, ge::DT_UINT64, ge::DT_INT32, ge::DT_UINT32, ge::DT_FLOAT, ge::DT_FLOAT16, | 29 | .DataType({ge::DT_INT64, ge::DT_UINT64, ge::DT_INT32, ge::DT_UINT32, ge::DT_FLOAT, ge::DT_FLOAT16, |
| 30 | - ge::DT_INT8, ge::DT_UINT8, ge::DT_BF16, ge::DT_INT16, ge::DT_UINT16, ge::DT_BOOL}) | 30 | + ge::DT_INT8, ge::DT_UINT8, ge::DT_BF16, ge::DT_INT16, ge::DT_UINT16, ge::DT_BOOL, ge::DT_COMPLEX32}) |
| 31 | .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 31 | .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, |
| 32 | - ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) | 32 | + ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 33 | .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 33 | .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, |
| 34 | - ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); | 34 | + ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); |
| 35 | this->Attr("num").AttrType(REQUIRED).Int(); | 35 | this->Attr("num").AttrType(REQUIRED).Int(); |
| 36 | this->Attr("axis").AttrType(OPTIONAL).Int(); | 36 | this->Attr("axis").AttrType(OPTIONAL).Int(); |
| 37 | 37 | ||