Searched defs:weight_vec (Results 1 – 2 of 2) sorted by relevance
68 auto weight_vec = fVec(weight_val); in lerp_scalar_kernel() local85 auto weight_vec = fVec(weight_val); in lerp_scalar_kernel() local122 [=](bVec self_vec, bVec end_vec, bVec weight_vec) -> bVec { in lerp_tensor_kernel()137 [=](hVec self_vec, hVec end_vec, hVec weight_vec) -> hVec { in lerp_tensor_kernel()
147 const __m256 weight_vec = _mm256_castps128_ps256(_mm_loadu_ps(weight)); in prepare_weights_for_2x2_convolve() local