Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_equalize.py300 def nn_module_supports_equalization(module) -> bool: function
316 nn_module_supports_equalization(modules[str(node.target)])
513 assert nn_module_supports_equalization(
688 assert nn_module_supports_equalization(module)
/aosp_15_r20/external/pytorch/torch/quantization/fx/
H A D_equalize.py29 nn_module_supports_equalization,