Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
Dlayer_norm_kernel.cu828 const int thr_load_row_off, in cuLoadWriteStridedInputs()
871 const int thr_load_row_off, in cuLoadAddStridedInputs()
920 const int thr_load_row_off = (threadIdx.x*blockDim.y)/blockDim.x + threadIdx.y*blockDim.y; in cuComputePartGradGammaBeta() local