Searched refs:QuantizedCustomModule (Results 1 – 1 of 1) sorted by relevance
886 class QuantizedCustomModule(torch.nn.Module): class958 ObservedCustomModule: QuantizedCustomModule977 self.assertEqual(type(m.sub.custom), QuantizedCustomModule)