Home
last modified time | relevance | path

Searched defs:vec_in (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/vec256/vsx/
H A Dvsx_helpers.h39 C10_ALWAYS_INLINE vfloat32 vec_float(const vint32& vec_in) { in vec_float()
47 C10_ALWAYS_INLINE vint32 vec_signed(const vfloat32& vec_in) { in vec_signed()
53 C10_ALWAYS_INLINE vint64 vec_signed(const vfloat64& vec_in) { in vec_signed()
61 C10_ALWAYS_INLINE vfloat32 vec_neg(const vfloat32& vec_in) { in vec_neg()
67 C10_ALWAYS_INLINE vfloat64 vec_neg(const vfloat64& vec_in) { in vec_neg()
73 C10_ALWAYS_INLINE vint16 vec_neg(const vint16& vec_in) { in vec_neg()
78 C10_ALWAYS_INLINE vint32 vec_neg(const vint32& vec_in) { in vec_neg()
83 C10_ALWAYS_INLINE vint64 vec_neg(const vint64& vec_in) { in vec_neg()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/
H A Dimpl.cpp210 float32x4x4_t vec_in = vld4q_f32(tmp_ptr + x); in neon_softmax_logits_1d_quantized() local
364 auto vec_in = wrapper::vloadq(tmp_ptr + x); in neon_softmax_logits_1d_float() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve/
H A Dimpl.cpp148 auto vec_in = svld1(pg, tmp_ptr + x); in sve_softmax_logits_1d_float() local