Searched refs:InferenceBatchSoftmax (Results 1 – 1 of 1) sorted by relevance
172 class InferenceBatchSoftmax(nn.Module): class336 self.inference_softmax = InferenceBatchSoftmax()