Home
last modified time | relevance | path

Searched refs:_set_root_node_getter (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/torch/ao/quantization/backend_config/
Donednn.py197 ._set_root_node_getter(_conv_bn_add_root_node_getter_left)
210 ._set_root_node_getter(_conv_add_root_node_getter_left)
283 ._set_root_node_getter(_conv_bn_add_root_node_getter_right)
296 ._set_root_node_getter(_conv_add_root_node_getter_right)
389 ._set_root_node_getter(_conv_bn_add_relu_root_node_getter_left)
402 ._set_root_node_getter(_conv_add_relu_root_node_getter_left)
485 ._set_root_node_getter(_conv_bn_add_relu_root_node_getter_right)
498 ._set_root_node_getter(_conv_add_relu_root_node_getter_right)
D_qnnpack_pt2e.py123 ._set_root_node_getter(root_node_getter)
Dbackend_config.py594 def _set_root_node_getter(self, root_node_getter: Callable) -> BackendPatternConfig: member in BackendPatternConfig
686 conf._set_root_node_getter(
DREADME.md130 * `_set_root_node_getter`
/external/pytorch/test/quantization/core/
Dtest_backend_config.py137 ._set_root_node_getter(_default_root_node_getter) \
213 conf._set_root_node_getter(_default_root_node_getter)
/external/pytorch/torch/ao/quantization/fx/
DREADME.md109 ._set_root_node_getter(my_root_node_getter)
118 `_set_root_node_getter`: sets a function that takes a node pattern and returns the root node in the…
/external/pytorch/test/quantization/fx/
Dtest_quantize_fx.py816 ._set_root_node_getter(conv_bn_res_relu_root_node_getter) \
882 ._set_root_node_getter(conv_res_relu_root_node_getter) \
6265 ._set_root_node_getter(root_node_getter)