!1723 fixed SoftmaxGradExt

Merge pull request !1723 from jiangjinsheng/vm_SoftmaxGradExt

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") \