Home
last modified time | relevance | path

Searched refs:dynamically_quantize_per_channel (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A Dmixtral_moe_quantize.py12 def dynamically_quantize_per_channel(x, quant_min, quant_max, target_dtype): function
82 int8_weight, scales, _ = dynamically_quantize_per_channel(
97 bit8_weight, scales, _ = dynamically_quantize_per_channel(
H A Dquantize.py10 def dynamically_quantize_per_channel(x, quant_min, quant_max, target_dtype): function
69 int8_weight, scales, _ = dynamically_quantize_per_channel(
/aosp_15_r20/external/executorch/examples/models/llama/tests/
H A Dtest_pre_quantization_transforms.py18 dynamically_quantize_per_channel,
104 weight_int8, scales, _ = dynamically_quantize_per_channel(
/aosp_15_r20/external/executorch/examples/models/llama/source_transformation/
H A Dquantize.py176 def dynamically_quantize_per_channel( function
367 weight, scales, _ = dynamically_quantize_per_channel(
590 weight, scales, _ = dynamically_quantize_per_channel(