update the registration info of InTopK operator.

4 years ago · 5ed6b320be
parent f7b5a327b8
commit 5ed6b320be
1 changed files with 0 additions and 1 deletions
--- a/mindspore/ops/_op_impl/tbe/in_top_k.py
+++ b/mindspore/ops/_op_impl/tbe/in_top_k.py
@ -23,7 +23,6 @@ in_top_k_op_info = TBERegOp("InTopK") \
    .compute_cost(10) \
    .kernel_name("in_top_k") \
    .partial_flag(True) \
-    .need_check_supported(True) \
    .attr("k", "required", "int", "all") \
    .input(0, "x1", False, "required", "all") \
    .input(1, "x2", False, "required", "all") \