Searched defs:warps_per_block (Results 1 – 3 of 3) sorted by relevance
44 int64_t warps_per_block = block / at::cuda::warp_size(); in compute_cuda() local
322 int warps_per_block = (threads_per_block / warp_size); in dispatch_softmax_forward() local372 int warps_per_block = (threads_per_block / warp_size); in dispatch_softmax_backward() local
531 const int warps_per_block = threads_per_block / at::cuda::warp_size(); in _embedding_bag_per_sample_weights_backward_cuda() local