Lines Matching refs:index

39 	uint32_t index;  in test_arm_max_f16()  local
42 arm_max_f16((float16_t *)input1, length, &val, &index); in test_arm_max_f16()
48 zassert_equal(index, ref_max_idx[ref_index], in test_arm_max_f16()
77 uint32_t index; in test_arm_min_f16() local
80 arm_min_f16((float16_t *)input1, length, &val, &index); in test_arm_min_f16()
86 zassert_equal(index, ref_min_idx[ref_index], in test_arm_min_f16()
98 uint32_t index; in test_arm_absmax_f16() local
101 arm_absmax_f16((float16_t *)input1, length, &val, &index); in test_arm_absmax_f16()
107 zassert_equal(index, ref_absmax_idx[ref_index], in test_arm_absmax_f16()
119 uint32_t index; in test_arm_absmin_f16() local
122 arm_absmin_f16((float16_t *)input1, length, &val, &index); in test_arm_absmin_f16()
128 zassert_equal(index, ref_absmin_idx[ref_index], in test_arm_absmin_f16()
303 size_t index; in ZTEST() local
317 for (index = 0; index < length; index++) { in ZTEST()
318 output[index] = in ZTEST()
319 arm_entropy_f16(input, in_entropy_dim[index + 1]); in ZTEST()
320 input += in_entropy_dim[index + 1]; in ZTEST()
338 size_t index; in ZTEST() local
352 for (index = 0; index < length; index++) { in ZTEST()
353 output[index] = in ZTEST()
354 arm_logsumexp_f16(input, in_logsumexp_dim[index + 1]); in ZTEST()
355 input += in_logsumexp_dim[index + 1]; in ZTEST()
373 size_t index; in ZTEST() local
388 for (index = 0; index < length; index++) { in ZTEST()
389 output[index] = in ZTEST()
391 input1, input2, in_kl_dim[index + 1]); in ZTEST()
393 input1 += in_kl_dim[index + 1]; in ZTEST()
394 input2 += in_kl_dim[index + 1]; in ZTEST()
413 size_t index; in ZTEST() local
432 for (index = 0; index < length; index++) { in ZTEST()
433 output[index] = in ZTEST()
436 in_logsumexp_dp_dim[index + 1], tmp); in ZTEST()
438 input1 += in_logsumexp_dp_dim[index + 1]; in ZTEST()
439 input2 += in_logsumexp_dp_dim[index + 1]; in ZTEST()