Searched refs:InnerModule2 (Results 1 – 1 of 1) sorted by relevance
11160 class InnerModule2(torch.nn.Module): class11163 self.weights = InnerModule2.get_embedding(embedding_dim)11193 self.module = InnerModule2(embedding_dim=8)