Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cpu/
Dgroup_norm_kernel.cpp593 opmath_t ds_val = std::accumulate(ds_arr.cbegin(), ds_arr.cend(), opmath_t(0)); in ComputeInternalGradients() local
685 opmath_t ds_val = std::accumulate(ds_arr.cbegin(), ds_arr.cend(), opmath_t(0)); in GroupNormInputBackward() local
1473 opmath_t ds_val{0}, db_val{0}; in GroupNormBackwardKernelImplChannelsLastInternal() local
1504 opmath_t ds_val = tmp_buffer_data[n * 2 * G + 2 * g]; in GroupNormBackwardKernelImplChannelsLastInternal() local