m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix some mistakes of TransData vm ops

Browse Source
pull/2501/head
liuwenhao4 5 years ago
parent 890e4cbf4a
commit 2f1cec7225
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/ops/_op_impl/tbe/trans_data.py
Unescape View File

@ -50,7 +50,7 @@ trans_data_op_info = TBERegOp("TransData") \
.dtype_format(DataType.F16_HWCN, DataType.F16_FracZ) \ .dtype_format(DataType.F16_HWCN, DataType.F16_FracZ) \
.dtype_format(DataType.F16_FracZ, DataType.F16_HWCN) \ .dtype_format(DataType.F16_FracZ, DataType.F16_HWCN) \
.dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_HWCN) \ .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_HWCN) \
.dtype_format(DataType.F16_HWCN, DataType.F16_5HD) \ .dtype_format(DataType.F16_HWCN, DataType.F16_C1HWNCoC0) \
.dtype_format(DataType.F16_Default, DataType.F16_FracNZ) \ .dtype_format(DataType.F16_Default, DataType.F16_FracNZ) \
.dtype_format(DataType.F32_Default, DataType.F32_FracNZ) \ .dtype_format(DataType.F32_Default, DataType.F32_FracNZ) \
.dtype_format(DataType.F16_FracNZ, DataType.F16_Default) \ .dtype_format(DataType.F16_FracNZ, DataType.F16_Default) \

Write Preview
Loading…
Cancel
Save