Searched defs:per_channel_zero_point (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantized.py | 196 def _fake_quantize_per_channel_affine_reference(X, per_channel_scale, per_channel_zero_point, axis,… argument
210 …tize_per_channel_affine_grad_reference(dY, X, per_channel_scale, per_channel_zero_point, axis, qua… argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_workflow_ops.py | 98 dY, X, per_channel_scale, per_channel_zero_point, axis, quant_min, quant_max, device): argument
|