Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_reshape.py207 class NetWithLoss5(nn.Cell): class
209 super(NetWithLoss5, self).__init__()
250 net = GradWrap5(NetWithLoss5(Net()))