Searched refs:checkGraphModuleNodes (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 324 self.checkGraphModuleNodes(
345 self.checkGraphModuleNodes(
372 self.checkGraphModuleNodes(
395 self.checkGraphModuleNodes(
414 self.checkGraphModuleNodes(
436 self.checkGraphModuleNodes(
454 self.checkGraphModuleNodes(
488 self.checkGraphModuleNodes(
516 self.checkGraphModuleNodes(
563 self.checkGraphModuleNodes(m, expected_node_occurrence=node_occurrence)
[all …]
|
| H A D | test_equalize_fx.py | 283 self.checkGraphModuleNodes(prepared, expected_node_occurrence=node_occurrence)
317 … self.checkGraphModuleNodes(prepared, expected_node_occurrence=no_eq_branching_node_occurrence)
341 self.checkGraphModuleNodes(prepared, expected_node_occurrence=eq_branching_node_occurrence)
790 self.checkGraphModuleNodes(equalized_quantized_model, expected_node_list=node_list)
896 self.checkGraphModuleNodes(equalized_model, expected_node_list=node_list)
|
| H A D | test_numeric_suite_fx.py | 885 self.checkGraphModuleNodes(
887 self.checkGraphModuleNodes(
889 self.checkGraphModuleNodes(
891 self.checkGraphModuleNodes(
953 self.checkGraphModuleNodes(
955 self.checkGraphModuleNodes(
1489 self.checkGraphModuleNodes(
1491 self.checkGraphModuleNodes(
1505 self.checkGraphModuleNodes(
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_quantize_pt2e.py | 201 self.checkGraphModuleNodes(
302 self.checkGraphModuleNodes(
405 self.checkGraphModuleNodes(
510 self.checkGraphModuleNodes(
596 self.checkGraphModuleNodes(
691 self.checkGraphModuleNodes(
825 self.checkGraphModuleNodes(
902 self.checkGraphModuleNodes(
1329 self.checkGraphModuleNodes(m, expected_node_occurrence=node_occurrence)
1346 self.checkGraphModuleNodes(m, expected_node_occurrence=node_occurrence)
[all …]
|
| H A D | test_representation.py | 46 self.checkGraphModuleNodes(model, expected_node_occurrence=ref_node_occurrence)
56 self.checkGraphModuleNodes(
|
| H A D | test_quantize_pt2e_qat.py | 428 self.checkGraphModuleNodes(
924 self.checkGraphModuleNodes(
1263 self.checkGraphModuleNodes(
|
| H A D | test_xnnpack_quantizer.py | 535 self.checkGraphModuleNodes(m, expected_node_occurrence=node_occurrence)
|
| H A D | test_x86inductor_quantizer.py | 575 self.checkGraphModuleNodes(
|
| /aosp_15_r20/external/executorch/backends/transforms/test/ |
| H A D | test_duplicate_dynamic_quant_chain.py | 74 self.checkGraphModuleNodes(m, expected_node_occurrence=node_occurrence)
79 self.checkGraphModuleNodes(m, expected_node_occurrence=node_occurrence)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 785 def checkGraphModuleNodes( member in QuantizationTestCase
1089 self.checkGraphModuleNodes(
1107 self.checkGraphModuleNodes(
1286 self.checkGraphModuleNodes(
1313 self.checkGraphModuleNodes(m_fx, expected_node_occurrence=node_occurrence)
|