Home
last modified time | relevance | path

Searched defs:fuse_conv_bn (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dfuser_method_mappings.py20 def fuse_conv_bn(is_qat, conv, bn): function
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpre_grad.py357 def fuse_conv_bn(gm: torch.fx.GraphModule, inplace=False) -> torch.fx.GraphModule: function