Searched defs:dim_block (Results 1 – 4 of 4) sorted by relevance
295 dim3 dim_block = cuda::getApplyBlock(); in tril_indices_cuda() local372 dim3 dim_block = cuda::getApplyBlock(); in triu_indices_cuda() local
229 dim3 dim_block) { in launcher()347 dim3 dim_block(block_size); in dropout_cuda() local
131 dim3 dim_block = cuda::getApplyBlock(); in roll_cuda() local
123 dim3 dim_block = block_size; in triu_tril_cuda_template() local