Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/ops/gpu/
Dtest_sgd_op.py29 class NetSGD(nn.Cell): class
31 super(NetSGD, self).__init__()
48 net = NetSGD()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_sgd_op.py29 class NetSGD(nn.Cell): class
31 super(NetSGD, self).__init__()
48 net = NetSGD()