Home
last modified time | relevance | path

Searched refs:SingleLayerFunctionalLinearModel (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_equalize_fx.py27 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 Dcommon_quantization.py1991 class SingleLayerFunctionalLinearModel(torch.nn.Module): class