Searched defs:gin_vec (Results 1 – 3 of 3) sorted by relevance
26 Vec gin_vec = Vec::loadu(gin + d) + Vec::loadu(gout + d); in nearest_channels_last_acc() local63 Vec gin_vec = Vec::loadu(gin + d) + Vec(w) * Vec::loadu(gout + d); in linear_channels_last_acc() local
349 Vec gin_vec = Vec::loadu(gin + d) + Vec::loadu(gout + d) / Vec(scalar_t(kh * kw)); in cpu_adaptive_avg_pool2d_backward_channels_last() local790 … Vec gin_vec = Vec::loadu(gin + d) + Vec::loadu(gout + d) / Vec(scalar_t(kd * kh * kw)); in cpu_adaptive_avg_pool3d_backward_channels_last() local
478 … Vec gin_vec = Vec::loadu(gin + d) + Vec::loadu(gout + d) / Vec(scalar_t(divide_factor)); in cpu_avg_pool2d_backward_channels_last() local1055 … Vec gin_vec = Vec::loadu(gin + d) + Vec::loadu(gout + d) / Vec(scalar_t(divide_factor)); in cpu_avg_pool3d_backward_channels_last() local