Searched refs:AnnotatedTwoLayerLinearModel (Results 1 – 3 of 3) sorted by relevance
32 AnnotatedTwoLayerLinearModel,256 model_list = [AnnotatedTwoLayerLinearModel()]
54 AnnotatedTwoLayerLinearModel,387 model = AnnotatedTwoLayerLinearModel()411 model = quantize(AnnotatedTwoLayerLinearModel(), test_only_eval_fn,
1622 class AnnotatedTwoLayerLinearModel(torch.nn.Module): class