Searched refs:TwoBlockComplexNet (Results 1 – 1 of 1) sorted by relevance
1268 class TwoBlockComplexNet(torch.nn.Module): class in TestFxDetectInputWeightEqualization1322 … non_fused = self._get_prepped_for_calibration_model(self.TwoBlockComplexNet(), detector_set)1323 …fused = self._get_prepped_for_calibration_model(self.TwoBlockComplexNet(), detector_set, fused=Tru…1366 model = self.TwoBlockComplexNet()