已合并
【bugfix】math仓conversion和random支持非连续 #1817
majiajian_hw创建于 3月23日
【bugfix】math仓conversion和random支持非连续 #1817
已合并
共 2 个文件变更+8-4
| @@ -29,7 +29,8 @@ public: | |||
| 29 | ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) | 29 | ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 30 | .UnknownShapeFormat( | 30 | .UnknownShapeFormat( |
| 31 | {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 31 | {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}) |
| 33 | + .AutoContiguous(); | ||
| 33 | this->Input("unique_indices") | 34 | this->Input("unique_indices") |
| 34 | .ParamType(REQUIRED) | 35 | .ParamType(REQUIRED) |
| 35 | .DataType( | 36 | .DataType( |
| @@ -40,7 +41,8 @@ public: | |||
| 40 | ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) | 41 | ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 41 | .UnknownShapeFormat( | 42 | .UnknownShapeFormat( |
| 42 | {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 43 | {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, |
| 43 | - ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); | 44 | + ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 45 | + .AutoContiguous(); | ||
| 44 | this->Input("indices") | 46 | this->Input("indices") |
| 45 | .ParamType(REQUIRED) | 47 | .ParamType(REQUIRED) |
| 46 | .DataType( | 48 | .DataType( |
| @@ -63,7 +65,8 @@ public: | |||
| 63 | ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) | 65 | ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 64 | .UnknownShapeFormat( | 66 | .UnknownShapeFormat( |
| 65 | {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, | 67 | {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, |
| 66 | - ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); | 68 | + ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 69 | + .AutoContiguous(); | ||
| 67 | this->Output("new_indices") | 70 | this->Output("new_indices") |
| 68 | .ParamType(REQUIRED) | 71 | .ParamType(REQUIRED) |
| 69 | .DataType( | 72 | .DataType( |
| @@ -24,7 +24,8 @@ public: | |||
| 24 | .ParamType(REQUIRED) | 24 | .ParamType(REQUIRED) |
| 25 | .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT}) | 25 | .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT}) |
| 26 | .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) | 26 | .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 27 | - .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}); | 27 | + .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND}) |
| 28 | + .AutoContiguous(); | ||
| 28 | this->Output("self") | 29 | this->Output("self") |
| 29 | .ParamType(REQUIRED) | 30 | .ParamType(REQUIRED) |
| 30 | .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT}) | 31 | .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT}) |