Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dgroup_norm_kernel.cpp609 const opmath_t* db_ptr, in CalcDsDb()
632 const opmath_t* db_ptr, in CalcDsDb()
682 const opmath_t* db_ptr = db + i * D; in GroupNormInputBackward() local
732 const PT* db_ptr = db + n * C + g * D + i; in GammaBackward() local
745 const PT* db_ptr = db + n * C + g * D + i; in GammaBackward() local
780 const opmath_t* db_ptr = db + n * C + g * D + i; in GammaBackward() local
797 const opmath_t* db_ptr = db + n * C + g * D + i; in GammaBackward() local
945 opmath_t* db_ptr, in DsDbRowwiseMomentsChannelsLast()
980 opmath_t* db_ptr, in DsDbRowwiseMomentsChannelsLast()
1246 opmath_t* db_ptr, in CalcInternalGradientsChannelsLast()
[all …]