Searched defs:shmem_sum_dy (Results 1 – 1 of 1) sorted by relevance
1139 T* shmem_sum_dy, in merge_block_vertical_backward()1248 static __shared__ accscalar_t shmem_sum_dy[MAX_BLOCK_SIZE]; in batch_norm_backward_reduce_channels_last_kernel() local