Searched refs:annotate_matmul (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/executorch/backends/qualcomm/quantizer/ |
| H A D | custom_annotation.py | 30 def annotate_matmul(node: Node, quantization_config: QuantizationConfig): function
101 annotate_matmul(node, quantization_config_16a8w)
110 def annotate_matmul(node: Node, quantization_config: QuantizationConfig): function
202 annotate_matmul(node, quantization_config_16a8w)
241 def annotate_matmul(node: Node, quantization_config: QuantizationConfig): function
259 annotate_matmul(node, quantization_config_16a8w)
|
| H A D | annotators.py | 689 def annotate_matmul(node: Node, quantization_config: QuantizationConfig) -> None: function
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/ |
| H A D | llama.py | 69 def annotate_matmul(node: Node, quantization_config: QuantizationConfig): function
138 annotate_matmul(node, quantization_config_16a8w)
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_x86inductor_quantizer.py | 2538 for annotate_matmul in [False, True]:
2554 if annotate_matmul:
2561 5 if annotate_matmul else 1
2564 7 if annotate_matmul else 3
2570 if annotate_matmul:
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_mkldnn_pattern_matcher.py | 2834 for annotate_matmul in [False, True]:
2849 3 if annotate_matmul else 0,
2854 if annotate_matmul:
|