Searched refs:row_idx (Results 1 – 3 of 3) sorted by relevance
68 int32_t row_idx; in arm_nn_softmax_common_s8() local70 for (row_idx = 0; row_idx < num_rows; ++row_idx) in arm_nn_softmax_common_s8()98 int16_t *output_s16 = (int16_t *)output + row_idx * row_size; in arm_nn_softmax_common_s8()121 int8_t *output_s8 = (int8_t *)output + row_idx * row_size; in arm_nn_softmax_common_s8()
55 int32_t row_idx; in arm_softmax_u8() local57 for (row_idx = 0; row_idx < num_rows; ++row_idx) in arm_softmax_u8()
48 int32_t row_idx; in arm_softmax_s16() local55 for (row_idx = 0; row_idx < num_rows; ++row_idx) in arm_softmax_s16()