Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/ops/gpu/
Dtest_gather_op.py26 class GatherNet(nn.Cell): class
28 super(GatherNet, self).__init__()
98 gather = GatherNet(dim)
110 gather = GatherNet(dim)
122 gather = GatherNet(dim)
134 gather = GatherNet(dim)
Dtest_gatherV2_op.py26 class GatherNet(nn.Cell): class
28 super(GatherNet, self).__init__()
842 gather = GatherNet()