!2351 Fix some mistakes of TransData vm ops

Merge pull request !2351 from liuwenhao/master
5 years ago · 108021b110
parent 9cddaee9b3 c444cb842f
commit 108021b110
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ops/_op_impl/tbe/trans_data.py
+++ b/mindspore/ops/_op_impl/tbe/trans_data.py
@ -58,7 +58,7 @@ trans_data_op_info = TBERegOp("TransData") \
    .dtype_format(DataType.F32_HWCN, DataType.F32_FracZ) \
    .dtype_format(DataType.F32_HWCN, DataType.F32_C1HWNCoC0) \
    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_HWCN) \
-    .dtype_format(DataType.F32_Default, DataType.F32_NCHW) \
+    .dtype_format(DataType.F32_Default, DataType.F32_HWCN) \
    .dtype_format(DataType.F32_HWCN, DataType.F32_Default) \
    .get_op_info()