Lines Matching refs:vidx
60 __m128i vidx = _mm_setzero_si128(); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4() local
64 vidx = _mm_or_si128(_mm_andnot_si128(vm1, vidx), _mm_and_si128(vm1, _mm_set1_epi32(1))); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
68 vidx = _mm_or_si128(_mm_andnot_si128(vm2, vidx), _mm_and_si128(vm2, _mm_set1_epi32(2))); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
72 vidx = _mm_or_si128(_mm_andnot_si128(vm3, vidx), _mm_and_si128(vm3, _mm_set1_epi32(3))); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
76 _mm_storeu_si128((__m128i*) index, vidx); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
86 __m128i vidx = _mm_setzero_si128(); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4() local
90 vidx = _mm_or_si128(_mm_andnot_si128(vm1, vidx), _mm_and_si128(vm1, _mm_set1_epi32(1))); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
94 vidx = _mm_or_si128(_mm_andnot_si128(vm2, vidx), _mm_and_si128(vm2, _mm_set1_epi32(2))); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
98 vidx = _mm_or_si128(_mm_andnot_si128(vm3, vidx), _mm_and_si128(vm3, _mm_set1_epi32(3))); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
102 _mm_storel_epi64((__m128i*) index, vidx); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
104 vidx = _mm_unpackhi_epi64(vidx, vidx); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
110 *index = (uint32_t) _mm_cvtsi128_si32(vidx); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()