!12263 update the registration info of InTopK operator.

From: @wangshuide2020
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui,@wuxuejian
pull/12263/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 7d5c44df90

@ -23,7 +23,6 @@ in_top_k_op_info = TBERegOp("InTopK") \
.compute_cost(10) \ .compute_cost(10) \
.kernel_name("in_top_k") \ .kernel_name("in_top_k") \
.partial_flag(True) \ .partial_flag(True) \
.need_check_supported(True) \
.attr("k", "required", "int", "all") \ .attr("k", "required", "int", "all") \
.input(0, "x1", False, "required", "all") \ .input(0, "x1", False, "required", "all") \
.input(1, "x2", False, "required", "all") \ .input(1, "x2", False, "required", "all") \

Loading…
Cancel
Save