Searched refs:nthread_y (Results 1 – 2 of 2) sorted by relevance
302 int nthread_y; in LayerNormGPUContig() local305 nthread_y = 1; in LayerNormGPUContig()307 nthread_y = 2; in LayerNormGPUContig()309 nthread_y = 4; in LayerNormGPUContig()312 const dim3 dimBlock(32, nthread_y); in LayerNormGPUContig()315 int nshared = nthread_y > 1 ? nthread_y * 32 * sizeof(AType) in LayerNormGPUContig()689 int nthread_y; in LayerNormGradGPUContig() local691 nthread_y = 1; in LayerNormGradGPUContig()693 nthread_y = 2; in LayerNormGradGPUContig()695 nthread_y = 4; in LayerNormGradGPUContig()[all …]