Searched refs:NetWithLossNoBias (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_reduce_method_info.py | 30 class NetWithLossNoBias(nn.Cell): class
32 super(NetWithLossNoBias, self).__init__()
206 net = GradWrapNoBias(NetWithLossNoBias(Net(strategy1, strategy2)))
229 net = GradWrapNoBias(NetWithLossNoBias(Net(strategy1, strategy2)))
252 net = GradWrapNoBias(NetWithLossNoBias(Net(strategy1, strategy2)))
462 net = GradWrapNoBias(NetWithLossNoBias(ArgMinWithValueNet2(strategy1, strategy2, strategy3)))
472 net = GradWrapNoBias(NetWithLossNoBias(ArgMinWithValueNet2(strategy1, strategy2, strategy3)))
482 net = GradWrapNoBias(NetWithLossNoBias(ArgMinWithValueNet2(strategy1, strategy2, strategy3)))
505 net = GradWrapNoBias(NetWithLossNoBias(Net(strategy1, strategy2, strategy3)))
531 net = GradWrapNoBias(NetWithLossNoBias(Net(strategy1, strategy2, strategy3)))
[all …]
|