From ed767b50a2f6898cd1c7ea39166b33085b65c5ea Mon Sep 17 00:00:00 2001 From: liangchenghui Date: Fri, 11 Dec 2020 11:08:30 +0800 Subject: [PATCH] Fix softmax dtype and format --- mindspore/ops/_op_impl/tbe/softmax.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/mindspore/ops/_op_impl/tbe/softmax.py b/mindspore/ops/_op_impl/tbe/softmax.py index 6d19c7f6ef..c27aa5c4ee 100644 --- a/mindspore/ops/_op_impl/tbe/softmax.py +++ b/mindspore/ops/_op_impl/tbe/softmax.py @@ -26,11 +26,8 @@ softmax_op_info = TBERegOp("Softmax") \ .attr("axis", "optional", "listInt", "all") \ .input(0, "x", False, "required", "all") \ .output(0, "y", False, "required", "all") \ - .dtype_format(DataType.F16_Default, DataType.F16_Default) \ - .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \ - .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ) \ - .dtype_format(DataType.F32_Default, DataType.F32_Default) \ - .dtype_format(DataType.F32_FracNZ, DataType.F32_FracNZ) \ + .op_pattern("dynamicFormat") \ + .dtype_format(DataType.None_None, DataType.None_None) \ .get_op_info()