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

!48653 Support new format for TransData 

Merge pull request !48653 from jiaorui/fix-transdata
This commit is contained in:
i-robot 2023-02-10 07:27:22 +00:00 committed by Gitee
parent ac441fc125 6f51f132f9
commit 39ea9099d9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/ops/_op_impl/tbe/trans_data_ds.py
View File

@ -83,6 +83,7 @@ trans_data_op_info = TBERegOp("TransData") \
.dtype_format(DataType.I8_NHWC, DataType.I8_HWCN) \
.dtype_format(DataType.I8_HWCN, DataType.I8_Default) \
.dtype_format(DataType.I8_HWCN, DataType.I8_NHWC) \
.dtype_format(DataType.I8_Default, DataType.I8_NC1HWC0) \
.dtype_format(DataType.I16_Default, DataType.I16_NHWC) \
.dtype_format(DataType.I16_Default, DataType.I16_HWCN) \
.dtype_format(DataType.I16_NHWC, DataType.I16_Default) \
@ -109,6 +110,7 @@ trans_data_op_info = TBERegOp("TransData") \
.dtype_format(DataType.U8_NHWC, DataType.U8_HWCN) \
.dtype_format(DataType.U8_HWCN, DataType.U8_Default) \
.dtype_format(DataType.U8_HWCN, DataType.U8_NHWC) \
.dtype_format(DataType.U8_Default, DataType.U8_NC1HWC0) \
.dtype_format(DataType.U16_Default, DataType.U16_NHWC) \
.dtype_format(DataType.U16_Default, DataType.U16_HWCN) \
.dtype_format(DataType.U16_NHWC, DataType.U16_Default) \

2
mindspore/python/mindspore/ops/op_info_register.py
View File

@ -1219,6 +1219,7 @@ class DataType:
I8_NDHWC = ("int8", "NDHWC")
I8_ChannelLast = ("int8", "ChannelLast")
I8_NDC1HWC0 = ("int8", "NDC1HWC0")
I8_NC1HWC0 = ("int8", "NC1HWC0")
U8_None = ("uint8", "")
U8_Default = ("uint8", "DefaultFormat")
@ -1232,6 +1233,7 @@ class DataType:
U8_NDHWC = ("uint8", "NDHWC")
U8_ChannelLast = ("uint8", "ChannelLast")
U8_NDC1HWC0 = ("uint8", "NDC1HWC0")
U8_NC1HWC0 = ("uint8", "NC1HWC0")
I16_None = ("int16", "")
I16_Default = ("int16", "DefaultFormat")