Home
last modified time | relevance | path

Searched refs:_sequential_wrapper2 (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dfuser_method_mappings.py181 def _sequential_wrapper2(sequential): function
201 (nn.Conv1d, nn.ReLU): _sequential_wrapper2(nni.ConvReLU1d),
202 (nn.Conv2d, nn.ReLU): _sequential_wrapper2(nni.ConvReLU2d),
203 (nn.Conv3d, nn.ReLU): _sequential_wrapper2(nni.ConvReLU3d),
205 (nn.Linear, nn.ReLU): _sequential_wrapper2(nni.LinearReLU),
206 (nn.BatchNorm2d, nn.ReLU): _sequential_wrapper2(nni.BNReLU2d),
207 (nn.BatchNorm3d, nn.ReLU): _sequential_wrapper2(nni.BNReLU3d),
/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A D_common_operator_config_utils.py15 _sequential_wrapper2,
216 .set_fuser_method(_sequential_wrapper2(nni.LinearReLU))
223 .set_fuser_method(_sequential_wrapper2(nni.LinearReLU))
331 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
338 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
699 .set_fuser_method(_sequential_wrapper2(fused_bn))
706 .set_fuser_method(_sequential_wrapper2(fused_bn))
H A Dexecutorch.py13 _sequential_wrapper2,
185 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
192 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
H A Donednn.py10 from torch.ao.quantization.fuser_method_mappings import _sequential_wrapper2
579 _sequential_wrapper2(nni.LinearLeakyReLU),
601 _sequential_wrapper2(nni.LinearTanh),
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_backend_config.py16 from torch.ao.quantization.fuser_method_mappings import _sequential_wrapper2
107 _fuser_method = _sequential_wrapper2(nni.LinearReLU)