Browse Source

!338 Fix arg_min_with_value op bug

Merge pull request !338 from zjun/fix_arg_min_with_value_bug
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
fa4a950562
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/tbe/arg_min_with_value.py

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

@@ -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…
Cancel
Save