fixed SoftmaxGradExt

5 years ago · 14587ad0a7
parent 0ac47f2f71
commit 14587ad0a7
1 changed files with 2 additions and 2 deletions
--- a/mindspore/ops/_op_impl/tbe/softmax_grad_ext.py
+++ b/mindspore/ops/_op_impl/tbe/softmax_grad_ext.py
@ -24,8 +24,8 @@ softmax_grad_ext_op_info = TBERegOp("SoftmaxGradExt") \
    .kernel_name("softmax_grad_ext") \
    .partial_flag(True) \
    .dynamic_format(True) \
-    .attr("axes", "required", "listInt", "all") \
-    .attr("keep_dims", "required", "bool", "all") \
+    .attr("axis", "required", "listInt", "all") \
+    .attr("keepdims", "required", "bool", "all") \
    .input(0, "grad", False, "required", "all") \
    .input(1, "x1", False, "required", "all") \
    .input(2, "x2", False, "required", "all") \