Home
last modified time | relevance | path

Searched defs:temp_vec1 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/executorch/kernels/optimized/blas/
H A DBlasKernel.cpp90 const bfloat16x8_t temp_vec1 = vld1q_bf16(reinterpret_cast<const __bf16*>( in dot_with_fp32_arith_main_inner_loop_bfdot() local
103 const uint16x8_t temp_vec1 = vld1q_u16(reinterpret_cast<const uint16_t*>( in dot_with_fp32_arith_main_inner_loop_no_bfdot() local
139 const auto temp_vec1 = in dot_with_fp32_arith_vectorized_tail_inner_loop() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBlasKernel.cpp432 const auto temp_vec1 = vld1q_f16(&vec1[registerPairIndex * 2 * kF32ElementsPerRegister]); in dot_with_fp32_arith_main_inner_loop() local
444 const auto temp_vec1 = vld1_f16(&vec1[idx]); in dot_with_fp32_arith_vectorized_tail_inner_loop() local
465 …const uint16x8_t temp_vec1 = vld1q_u16(reinterpret_cast<const uint16_t*>(&vec1[registerPairIndex *… in dot_with_fp32_arith_main_inner_loop() local
477 const auto temp_vec1 = vld1_u16(reinterpret_cast<const uint16_t*>(&vec1[idx])); in dot_with_fp32_arith_vectorized_tail_inner_loop() local