Searched defs:NetArgmax (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ | ||
| D | test_argmax_op.py | 28 class NetArgmax(nn.Cell): class |
| /third_party/mindspore/tests/st/ops/cpu/ | ||
| D | test_argmax_op.py | 30 class NetArgmax(nn.Cell): class |