From c444cb842f83cd22a86c09600a2594500242de42 Mon Sep 17 00:00:00 2001 From: liuwenhao4 Date: Fri, 19 Jun 2020 15:33:33 +0800 Subject: [PATCH] Fix some mistakes of TransData vm ops --- mindspore/ops/_op_impl/tbe/trans_data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/_op_impl/tbe/trans_data.py b/mindspore/ops/_op_impl/tbe/trans_data.py index 64cca3dc44..da5ae6e1b8 100644 --- 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()