Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py11160 class InnerModule2(torch.nn.Module): class
11163 self.weights = InnerModule2.get_embedding(embedding_dim)
11193 self.module = InnerModule2(embedding_dim=8)