Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_equalize_fx.py28 TwoLayerFunctionalLinearModel,
261 (TwoLayerFunctionalLinearModel, two_F_layer_node_occurrence),
351 (TwoLayerFunctionalLinearModel, 2),
423 SingleLayerFunctionalLinearModel, TwoLayerFunctionalLinearModel]
475 SingleLayerFunctionalLinearModel, TwoLayerFunctionalLinearModel]
766 (TwoLayerFunctionalLinearModel, functionalLinear2_node_list),
799 SingleLayerFunctionalLinearModel, TwoLayerFunctionalLinearModel]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py2003 class TwoLayerFunctionalLinearModel(torch.nn.Module): class