Home
last modified time | relevance | path

Searched defs:QuantizationConfig (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/qualcomm/quantizer/
H A Dannotators.py86 node: Node, quantization_config: QuantizationConfig
114 node: Node, quantization_config: QuantizationConfig
133 def annotate_topk(node: Node, quantization_config: QuantizationConfig) -> None:
140 def annotate_binary(node: Node, quantization_config: QuantizationConfig) -> None:
166 def annotate_add(node: Node, quantization_config: QuantizationConfig) -> None:
171 def annotate_sub(node: Node, quantization_config: QuantizationConfig) -> None:
178 def annotate_mul(node: Node, quantization_config: QuantizationConfig) -> None:
185 def annotate_div(node: Node, quantization_config: QuantizationConfig) -> None:
250 def annotate_rsub(node: Node, quantization_config: QuantizationConfig) -> None:
255 def annotate_sum(node: Node, quantization_config: QuantizationConfig) -> None:
[all …]
H A Dcustom_annotation.py30 def annotate_matmul(node: Node, quantization_config: QuantizationConfig):
46 def annotate_cat(node: Node, quantization_config: QuantizationConfig):
67 node: Node, quantization_config: QuantizationConfig
110 def annotate_matmul(node: Node, quantization_config: QuantizationConfig):
124 def annotate_index_put(node: Node, quantization_config: QuantizationConfig) -> None:
137 node: Node, quantization_config: QuantizationConfig
148 def annotate_cat(node: Node, quantization_config: QuantizationConfig):
174 def annotate_matmul_input1(node: Node, quantization_config: QuantizationConfig):
207 def annotate_conv2d(node: Node, quantization_config: QuantizationConfig) -> None:
241 def annotate_matmul(node: Node, quantization_config: QuantizationConfig):
H A Dqconfig.py21 class QuantizationConfig: class
/aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/
H A Dx86_inductor_quantizer.py404 quantization_config: Optional["QuantizationConfig"],
456 cls, quantization_config: Optional[QuantizationConfig]
506 self, quantization_config: Optional[QuantizationConfig]
537 def set_global(self, quantization_config: QuantizationConfig):
556 quantization_config: Optional[QuantizationConfig],
575 quantization_config: Optional[QuantizationConfig],
590 self, module_name: str, quantization_config: Optional[QuantizationConfig]
604 quantization_config: Optional[QuantizationConfig],
618 quantization_config: Optional[QuantizationConfig],
650 quantization_config: Optional[QuantizationConfig],
[all …]
H A Dxnnpack_quantizer_utils.py51 class QuantizationConfig: class
119 def get_input_act_qspec(quantization_config: Optional[QuantizationConfig]):
132 def get_output_act_qspec(quantization_config: Optional[QuantizationConfig]):
145 def get_weight_qspec(quantization_config: Optional[QuantizationConfig]):
162 def get_bias_qspec(quantization_config: Optional[QuantizationConfig]):
178 quantization_config: Optional[QuantizationConfig],
226 quantization_config: Optional[QuantizationConfig],
288 quantization_config: Optional[QuantizationConfig],
335 quantization_config: Optional[QuantizationConfig],
391 quantization_config: Optional[QuantizationConfig],
[all …]
H A Dxnnpack_quantizer.py287 cls, quantization_config: Optional[QuantizationConfig]
305 def set_global(self, quantization_config: QuantizationConfig) -> XNNPACKQuantizer:
312 quantization_config: QuantizationConfig,
318 self, module_type: Callable, quantization_config: QuantizationConfig
328 self, module_name: str, quantization_config: Optional[QuantizationConfig]
359 quantization_config: Optional[QuantizationConfig],
376 quantization_config: Optional[QuantizationConfig],
H A Dembedding_quantizer.py57 cls, quantization_config: QuantizationConfig
/aosp_15_r20/external/executorch/backends/mediatek/quantizer/
H A Dannotator.py31 def annotate(graph: Graph, quant_config: QuantizationConfig) -> None:
84 graph: Graph, quant_config: QuantizationConfig
149 def _annotate_rmsnorm_pattern(graph: Graph, quant_config: QuantizationConfig) -> None:
182 def annotate_placeholder(node: Node, quant_config: QuantizationConfig) -> None:
199 def annotate_affine_ops(node: Node, quant_config: QuantizationConfig) -> None:
250 def annotate_output_qspec(node: Node, quant_config: QuantizationConfig) -> None:
258 def annotate_embedding_op(node: Node, quant_config: QuantizationConfig) -> None:
/aosp_15_r20/external/executorch/backends/arm/quantizer/
H A Darm_quantizer.py283 def set_global(self, quantization_config: QuantizationConfig) -> ArmQuantizer:
289 self, module_type: Callable, quantization_config: QuantizationConfig
299 self, module_name: str, quantization_config: Optional[QuantizationConfig]
338 quantization_config: Optional[QuantizationConfig],
390 quantization_config: QuantizationConfig,
424 cls, quantization_config: Optional[QuantizationConfig]
H A Dquantization_config.py20 class QuantizationConfig: class
H A Darm_quantizer_utils.py61 node: Node, gm: GraphModule, quantization_config: QuantizationConfig
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/
H A Dllama.py69 def annotate_matmul(node: Node, quantization_config: QuantizationConfig):
85 def annotate_cat(node: Node, quantization_config: QuantizationConfig):
106 node: Node, quantization_config: QuantizationConfig
151 def annotate_conv2d(node: Node, quantization_config: QuantizationConfig) -> None:
/aosp_15_r20/external/executorch/backends/cadence/aot/quantizer/
H A Dquantizer.py66 self, pattern: QuantizationPattern, quantization_config: QuantizationConfig
145 self, quantization_config: Optional[QuantizationConfig] = None
/aosp_15_r20/external/executorch/backends/vulkan/quantizer/
H A Dvulkan_quantizer.py81 def set_global(self, quantization_config: QuantizationConfig) -> VulkanQuantizer:
100 quantization_config: Optional[QuantizationConfig],
/aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/
H A Dlinear_annotator.py25 quantization_config: QuantizationConfig,
H A Dsub_annotator.py22 quantization_config: QuantizationConfig,
H A Dconv_annotator.py23 quantization_config: QuantizationConfig,
H A Dmm_annotator.py24 quantization_config: QuantizationConfig,
H A Dmax_pool2d_annotator.py27 quantization_config: QuantizationConfig,
H A Dadd_annotator.py22 quantization_config: QuantizationConfig,
H A Done_to_one_annotator.py26 quantization_config: QuantizationConfig,
H A Dmul_annotator.py23 quantization_config: QuantizationConfig,
H A Dgeneric_annotator.py65 quantization_config: QuantizationConfig,
H A Dadaptive_ang_pool2d_annotator.py28 quantization_config: QuantizationConfig,
H A Dupsample_nearest2d_annotator.py36 quantization_config: QuantizationConfig,

12