Lines Matching refs:vidx
56 uint32x4_t vidx = vmovq_n_u32(0); in xnn_f32_argmaxpool_ukernel_4x__neon_c4() local
60 vidx = vbslq_u32(vm1, vmovq_n_u32(1), vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
64 vidx = vbslq_u32(vm2, vmovq_n_u32(2), vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
68 vidx = vbslq_u32(vm3, vmovq_n_u32(3), vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
71 vst1q_u32(index, vidx); index += 4; in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
80 uint32x4_t vidx = vmovq_n_u32(0); in xnn_f32_argmaxpool_ukernel_4x__neon_c4() local
84 vidx = vbslq_u32(vm1, vmovq_n_u32(1), vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
88 vidx = vbslq_u32(vm2, vmovq_n_u32(2), vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
92 vidx = vbslq_u32(vm3, vmovq_n_u32(3), vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
95 uint32x2_t vidx_lo = vget_low_u32(vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
100 vidx_lo = vget_high_u32(vidx); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()