From ad3f78a4055f4e418b9fd7cb0f3963c8311846ae Mon Sep 17 00:00:00 2001 From: liuwenhao4 Date: Sun, 28 Jun 2020 09:47:12 +0800 Subject: [PATCH] Fix some mistakes of ConfusionTransposeD vm ops --- .../ops/_op_impl/tbe/confusion_transpose_d.py | 22 ++----------------- 1 file changed, 2 insertions(+), 20 deletions(-) diff --git a/mindspore/ops/_op_impl/tbe/confusion_transpose_d.py b/mindspore/ops/_op_impl/tbe/confusion_transpose_d.py index e52ae01520b..f84e9d4292d 100644 --- a/mindspore/ops/_op_impl/tbe/confusion_transpose_d.py +++ b/mindspore/ops/_op_impl/tbe/confusion_transpose_d.py @@ -28,26 +28,8 @@ confusion_transpose_d_op_info = TBERegOp("ConfusionTransposeD") \ .attr("transpose_first", "required", "bool", "all") \ .input(0, "x", False, "required", "all") \ .output(0, "y", False, "required", "all") \ - .dtype_format(DataType.I8_FracNZ, DataType.I8_FracNZ) \ - .dtype_format(DataType.I8_Default, DataType.I8_Default) \ - .dtype_format(DataType.U8_FracNZ, DataType.U8_FracNZ) \ - .dtype_format(DataType.U8_Default, DataType.U8_Default) \ - .dtype_format(DataType.I16_FracNZ, DataType.I16_FracNZ) \ - .dtype_format(DataType.I16_Default, DataType.I16_Default) \ - .dtype_format(DataType.U16_FracNZ, DataType.U16_FracNZ) \ - .dtype_format(DataType.U16_Default, DataType.U16_Default) \ - .dtype_format(DataType.I32_FracNZ, DataType.I32_FracNZ) \ - .dtype_format(DataType.I32_Default, DataType.I32_Default) \ - .dtype_format(DataType.U32_FracNZ, DataType.U32_FracNZ) \ - .dtype_format(DataType.U32_Default, DataType.U32_Default) \ - .dtype_format(DataType.I64_FracNZ, DataType.I64_FracNZ) \ - .dtype_format(DataType.I64_Default, DataType.I64_Default) \ - .dtype_format(DataType.U64_FracNZ, DataType.U64_FracNZ) \ - .dtype_format(DataType.U64_Default, DataType.U64_Default) \ - .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ) \ - .dtype_format(DataType.F16_Default, DataType.F16_Default) \ - .dtype_format(DataType.F32_FracNZ, DataType.F32_FracNZ) \ - .dtype_format(DataType.F32_Default, DataType.F32_Default) \ + .op_pattern("dynamicFormat") \ + .dtype_format(DataType.None_None, DataType.None_None) \ .get_op_info()