Browse Source

update the registration info of InTopK operator.

tags/v1.2.0-rc1
wangshuide2020 4 years ago
parent
commit
5ed6b320be
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      mindspore/ops/_op_impl/tbe/in_top_k.py

+ 0
- 1
mindspore/ops/_op_impl/tbe/in_top_k.py View File

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


Loading…
Cancel
Save