Searched defs:val_vec (Results 1 – 5 of 5) sorted by relevance
81 [found_inf_ptr, inv_scale_ptr](Vectorized<scalar_t> val_vec) -> Vectorized<scalar_t>{ in _amp_foreach_non_finite_check_and_unscale_cpu_kernel()111 [found_inf_ptr, inv_scale_ptr](Vectorized<scalar_t> val_vec) -> Vectorized<scalar_t>{ in _amp_foreach_non_finite_check_and_unscale_cpu_kernel()
160 Vec val_vec = Vec::loadu(in + d2); in cpu_adaptive_max_pool2d_channels_last() local641 Vec val_vec = Vec::loadu(in + d2); in cpu_adaptive_max_pool3d_channels_last() local
35 aVec val_vec = aVec((opmath_t)val); in _update() local
111 Vec val_vec = Vec::loadu(in + d2); in compute_internal() local
1947 std::vector<uint8_t> val_vec(reinterpret_cast<uint8_t*>(p_value), in metadataChangedNative() local