Searched refs:row_size (Results 1 – 3 of 3) sorted by relevance
88 const int32_t row_size, in arm_softmax_s8() argument105 int32_t vec_count = (row_size + 15) / 16; in arm_softmax_s8()106 uint32_t r_count = (uint32_t)row_size; in arm_softmax_s8()115 vec_count = row_size / 4; in arm_softmax_s8()140 const int32_t tail_idx = row_size & ~3; in arm_softmax_s8()141 for (int i = 0; i < (row_size & 3); i++) in arm_softmax_s8()154 vec_count = row_size / 4; in arm_softmax_s8()189 for (int i = 0; i < (row_size & 3); i++) in arm_softmax_s8()204 input += row_size; in arm_softmax_s8()205 output += row_size; in arm_softmax_s8()[all …]
46 const int32_t row_size, in arm_softmax_u8() argument62 for (col = 1; col < row_size; ++col) in arm_softmax_u8()70 for (col = 0; col < row_size; ++col) in arm_softmax_u8()83 for (col = 0; col < row_size; ++col) in arm_softmax_u8()97 input += row_size; in arm_softmax_u8()98 output += row_size; in arm_softmax_u8()
1756 const int32_t row_size,1779 const int32_t row_size,
