p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify_version info for 1.1.0

This commit is contained in:
lilei 2020-10-29 19:57:20 +08:00
parent 8f44074796
commit 740d5bac67
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphengine

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

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