Searched refs:GatherNetDynamic (Results 1 – 1 of 1) sorted by relevance
944 class GatherNetDynamic(nn.Cell): class946 super(GatherNetDynamic, self).__init__()970 gather = GatherNetDynamic()1001 gather = GatherNetDynamic(-1, True, False)1077 gather = GatherNetDynamic(-1, False, True)