Home
last modified time | relevance | path

Searched refs:NetArgmaxWithValueBig (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/tests/st/ops/cpu/
Dtest_argmaxwithvalue_op.py38 class NetArgmaxWithValueBig(nn.Cell): class
40 super(NetArgmaxWithValueBig, self).__init__()
82 argmax = NetArgmaxWithValueBig(0)
89 argmax = NetArgmaxWithValueBig(1)
96 argmax = NetArgmaxWithValueBig(2)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_argmaxwithvalue_op.py38 class NetArgmaxWithValueBig(nn.Cell): class
40 super(NetArgmaxWithValueBig, self).__init__()
82 argmax = NetArgmaxWithValueBig(0)
89 argmax = NetArgmaxWithValueBig(1)
96 argmax = NetArgmaxWithValueBig(2)