Searched refs:ReferenceQuantizedModule (Results 1 – 5 of 5) sorted by relevance
8 from .utils import ReferenceQuantizedModule14 class Embedding(nn.Embedding, ReferenceQuantizedModule):82 class EmbeddingBag(nn.EmbeddingBag, ReferenceQuantizedModule):
8 from .utils import ReferenceQuantizedModule14 class Linear(nn.Linear, ReferenceQuantizedModule):
9 from .utils import ReferenceQuantizedModule22 class _ConvNd(torch.nn.modules.conv._ConvNd, ReferenceQuantizedModule):
12 class ReferenceQuantizedModule(torch.nn.Module): class
17 ReferenceQuantizedModule,