Browse Source

modify_version info for 1.1.0

tags/v1.1.0
lilei 5 years ago
parent
commit
740d5bac67
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      graphengine
  2. +2
    -2
      mindspore/ops/_op_impl/tbe/top_k.py

+ 1
- 1
graphengine

@@ -1 +1 @@
Subproject commit 433521ddbf33918a86d0a16bafa200c95baa4eea
Subproject commit 423c0228e8c421f2b095e40d14e9fb3b563f63aa

+ 2
- 2
mindspore/ops/_op_impl/tbe/top_k.py View File

@@ -19,9 +19,9 @@ from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
top_k_op_info = TBERegOp("TopK") \
.fusion_type("OPAQUE") \
.async_flag(False) \
.binfile_name("top_k.so") \
.binfile_name("top_k_d.so") \
.compute_cost(10) \
.kernel_name("top_k") \
.kernel_name("top_k_d") \
.partial_flag(True) \
.attr("dim", "optional", "int", "all") \
.attr("k", "required", "int", "all") \


Loading…
Cancel
Save