Searched refs:ngrid_y (Results 1 – 2 of 2) sorted by relevance
301 int ngrid_y = (nbatch > kMaxGridDim) ? kBaseGridNum : 1; in LayerNormGPUContig() local303 const dim3 dimGrid(ngrid_x, ngrid_y); in LayerNormGPUContig()687 int ngrid_y = (nbatch > kMaxGridDim) ? kBaseGridNum : 1; in LayerNormGradGPUContig() local688 const dim3 data_grid_dim(ngrid_x, ngrid_y); in LayerNormGradGPUContig()