Searched refs:SingleLayerFunctionalLinearModel (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_equalize_fx.py | 27 SingleLayerFunctionalLinearModel,
350 (SingleLayerFunctionalLinearModel, 2), (FunctionalLinearAddModel, 2),
423 SingleLayerFunctionalLinearModel, TwoLayerFunctionalLinearModel]
475 SingleLayerFunctionalLinearModel, TwoLayerFunctionalLinearModel]
532 tests = [SingleLayerLinearModel, TwoLayerLinearModel, SingleLayerFunctionalLinearModel]
764 (SingleLayerFunctionalLinearModel, functionalLinear_node_list),
799 SingleLayerFunctionalLinearModel, TwoLayerFunctionalLinearModel]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1991 class SingleLayerFunctionalLinearModel(torch.nn.Module): class
|