Searched refs:AttentionAggregator (Results 1 – 3 of 3) sorted by relevance
19 from aggregator import AttentionAggregator85 self.layers.append(AttentionAggregator(95 self.layers.append(AttentionAggregator(104 self.layers.append(AttentionAggregator(
17 from aggregator import MeanAggregator, AttentionHead, AttentionAggregator75 net = AttentionAggregator(1433, 8, 8)
328 class AttentionAggregator(nn.Cell): class366 super(AttentionAggregator, self).__init__()