From 5ed6b320beb23c44edb51fcfc180dbf5f8133a3f Mon Sep 17 00:00:00 2001 From: wangshuide2020 <7511764+wangshuide2020@user.noreply.gitee.com> Date: Mon, 8 Feb 2021 11:13:38 +0800 Subject: [PATCH] update the registration info of InTopK operator. --- mindspore/ops/_op_impl/tbe/in_top_k.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mindspore/ops/_op_impl/tbe/in_top_k.py b/mindspore/ops/_op_impl/tbe/in_top_k.py index dc87df21cd..46d7258e2a 100644 --- 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") \