Searched defs:test_argmax_1d (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ | ||
| D | test_argmax_op.py | 40 def test_argmax_1d(): function |
| /third_party/mindspore/tests/st/ops/cpu/ | ||
| D | test_argmax_op.py | 42 def test_argmax_1d(): function |