Home
last modified time | relevance | path

Searched refs:set_fuser_method (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A Donednn.py196 .set_fuser_method(_fuse_conv_bn_add_left)
209 .set_fuser_method(_fuse_conv_add_left)
282 .set_fuser_method(_fuse_conv_bn_add_right)
295 .set_fuser_method(_fuse_conv_add_right)
388 .set_fuser_method(_fuse_conv_bn_add_relu_left)
401 .set_fuser_method(_fuse_conv_add_relu_left)
484 .set_fuser_method(_fuse_conv_bn_add_relu_right)
497 .set_fuser_method(_fuse_conv_add_relu_right)
538 .set_fuser_method(fuser_method)
545 .set_fuser_method(fuser_method)
[all …]
H A D_common_operator_config_utils.py216 .set_fuser_method(_sequential_wrapper2(nni.LinearReLU))
223 .set_fuser_method(_sequential_wrapper2(nni.LinearReLU))
265 .set_fuser_method(fuse_linear_bn)
331 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
338 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
394 .set_fuser_method(fuse_conv_bn)
401 .set_fuser_method(fuse_conv_bn_relu)
409 .set_fuser_method(fuse_conv_bn_relu)
459 .set_fuser_method(fuse_convtranspose_bn)
699 .set_fuser_method(_sequential_wrapper2(fused_bn))
[all …]
H A Dexecutorch.py185 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
192 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))
244 .set_fuser_method(fuse_conv_bn)
251 .set_fuser_method(fuse_conv_bn_relu)
259 .set_fuser_method(fuse_conv_bn_relu)
H A DREADME.md94 .set_fuser_method(fuse_conv2d_relu)
128 * `set_fuser_method`
133 …ed during the prepare phase according to the function specified in `set_fuser_method`, which repla…
H A Dbackend_config.py576 def set_fuser_method(self, fuser_method: Callable) -> BackendPatternConfig: member in BackendPatternConfig
683 conf.set_fuser_method(
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_backend_config.py132 .set_fuser_method(self._fuser_method)
207 conf.set_fuser_method(self._fuser_method)
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A DREADME.md108 .set_fuser_method(fuse_linear_relu)
117 `set_fuser_method`: specifies the fuser method to use for the pattern, a fuser method will take the…
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py753 .set_fuser_method(fuse_conv_bn_relu)
756 .set_fuser_method(fuse_conv_bn_relu)
815 .set_fuser_method(fuse_conv_bn_relu) \
878 .set_fuser_method(fuse_conv_relu)
881 .set_fuser_method(fuse_conv_res_relu) \