Searched defs:NetSoftmax (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ | ||
| D | test_softmax_op.py | 29 class NetSoftmax(nn.Cell): class |
| /third_party/mindspore/tests/st/ops/gpu/ | ||
| D | test_softmax_op.py | 26 class NetSoftmax(nn.Cell): class |