Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/python/parallel/
Dtest_add_relu_redistribution.py29 class AddRelu(nn.Cell): class
31 super(AddRelu, self).__init__()
72 net = Grad(NetWithLoss(AddRelu(strategy0, strategy1)))
85 net = Grad(NetWithLoss(AddRelu(strategy0, strategy1)))