Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_equalize.py290 def fused_module_supports_equalization(module) -> bool: function
317 or fused_module_supports_equalization(modules[str(node.target)])
509 if fused_module_supports_equalization(modules[str(node.target)]):
686 if fused_module_supports_equalization(modules[str(op_node.target)]):
/aosp_15_r20/external/pytorch/torch/quantization/fx/
H A D_equalize.py20 fused_module_supports_equalization,