Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp400 int overflow_threshold = 262144; // 2147483647(max of int32)/(256*256)*8 = 262144 in hsum_sq() local
401 int loop = len / overflow_threshold + 1; in hsum_sq()
403 for (; ((i < overflow_threshold * j) && (i < len / 16 * 16)); i += 16) { in hsum_sq()
429 int overflow_threshold = 262144; // 2147483647(max of int32)/(512*512)*8 = 262144 in hsum_sq() local
430 int loop = len / overflow_threshold + 1; in hsum_sq()
432 for (; ((i < overflow_threshold * j) && (i < len / 32 * 32)); i += 32) { in hsum_sq()
475 int overflow_threshold = 1048576; //2147483647/(128*128)*8 = 1048576 in hsum_sq() local
476 int loop = len / overflow_threshold + 1; in hsum_sq()
479 for (; ((i < overflow_threshold * j) && (i < len / 16 * 16)); i += 16) { in hsum_sq()
508 int overflow_threshold = 1048576; //2147483647/(256*256)*8 = 1048576 in hsum_sq() local
[all …]