Home
last modified time | relevance | path

Searched refs:node_supports_equalization (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_equalize.py310 def node_supports_equalization(node: Node, modules) -> bool: function
349 if node_supports_equalization(user, modules):
418 assert node_supports_equalization(node, modules)
770 node_supports_equalization(prev_node, modules)
H A Dprepare.py50 from ._equalize import is_equalization_observer, node_supports_equalization
1042 if equalization_qconfig is None or not node_supports_equalization(
/aosp_15_r20/external/pytorch/torch/quantization/fx/
H A D_equalize.py30 node_supports_equalization,