Searched defs:dy_vec (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | group_norm_kernel.cpp | 955 Vec dy_vec = Vec::loadu(dY_ptr + d); in DsDbRowwiseMomentsChannelsLast() local
966 Vec dy_vec = Vec::loadu(dY_ptr + d, C - d); in DsDbRowwiseMomentsChannelsLast() local
991 Vec dy_vec = Vec::loadu(dY_ptr + d); in DsDbRowwiseMomentsChannelsLast() local
1011 Vec dy_vec = Vec::loadu(dY_ptr + d, C - d); in DsDbRowwiseMomentsChannelsLast() local
1075 auto dy_vec = vec::Vectorized<T>::loadu(dY_ptr + d); in ApplyInputGradientsChannelsLastColMov() local
1090 auto dy_vec = vec::Vectorized<T>::loadu(dY_ptr + d, D - d); in ApplyInputGradientsChannelsLastColMov() local
1127 Vec dy_vec = Vec::loadu(dY_ptr + d); in ApplyInputGradientsChannelsLastColMov() local
1145 Vec dy_vec = Vec::loadu(dY_ptr + d, D - d); in ApplyInputGradientsChannelsLastColMov() local
1175 auto dy_vec = vec::Vectorized<T>::loadu(dY_data + d); in ApplyInputGradientsChannelsLastRowMov() local
1184 auto dy_vec = vec::Vectorized<T>::loadu(dY_data + d, D - d); in ApplyInputGradientsChannelsLastRowMov() local
[all …]
|
| H A D | Activation.cpp | 470 [&](Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) -> Vectorized<scalar_t> { in GeluBackwardKernelImpl()
533 [&](Vec dy_vec, Vec x_vec) { in GeluBackwardKernelImpl()
605 [&](Vec dy_vec, Vec x_vec) { in GeluBackwardKernelImpl()
1264 [kOneVec](Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) -> Vectorized<scalar_t> { in silu_backward_kernel()
1287 [kOneVec](Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) { in silu_backward_kernel()
1340 [kOneVec](Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) -> Vectorized<scalar_t> { in mish_backward_kernel()
1365 [kOneVec](Vec dy_vec, Vec x_vec) -> Vec { in mish_backward_kernel()
|
| H A D | BinaryOpsKernel.cpp | 875 Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) { in logit_backward_kernel()
896 Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) { in logit_backward_kernel()
|