forked from mindspore-Ecosystem/mindspore
Fix arg_min_with_value bug
This commit is contained in:
parent
f7aadb4da8
commit
5617143fb8
|
@ -16,7 +16,7 @@
|
|||
"""ArgMinWithValue op"""
|
||||
from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
|
||||
|
||||
arg_min_with_value_op_info = TBERegOp("ArgMaxWithValue") \
|
||||
arg_min_with_value_op_info = TBERegOp("ArgMinWithValue") \
|
||||
.fusion_type("ELEMWISE") \
|
||||
.async_flag(False) \
|
||||
.binfile_name("arg_min_with_value.so") \
|
||||
|
|
Loading…
Reference in New Issue