Home
last modified time | relevance | path

Searched defs:NetSoftmax (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/tests/st/ops/cpu/
Dtest_softmax_op.py29 class NetSoftmax(nn.Cell): class
/third_party/mindspore/tests/st/ops/gpu/
Dtest_softmax_op.py26 class NetSoftmax(nn.Cell): class