From 2f1cec7225698a967f1285c1b3f1d26a30ea47fa 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 c0cce302c..91b4bc85f 100644 --- a/mindspore/ops/_op_impl/tbe/trans_data.py +++ b/mindspore/ops/_op_impl/tbe/trans_data.py @@ -50,7 +50,7 @@ trans_data_op_info = TBERegOp("TransData") \ .dtype_format(DataType.F16_HWCN, DataType.F16_FracZ) \ .dtype_format(DataType.F16_FracZ, 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.F32_Default, DataType.F32_FracNZ) \ .dtype_format(DataType.F16_FracNZ, DataType.F16_Default) \ -- GitLab