Searched defs:blocks_trans (Results 1 – 1 of 1) sorted by relevance
726 dim3 blocks_trans(input.size(1), std::max<int>(1, std::min<int>((256*1024)/input.size(1), in batch_norm_elemt_cuda_template() local874 dim3 blocks_trans(input.size(1), std::max<int>(1, std::min<int>((256*1024)/input.size(1), in batch_norm_backward_elemt_cuda_template() local925 dim3 blocks_trans(input.size(1), std::max<int>(1, std::min<int>((256*1024)/input.size(1), in batch_norm_backward_elemt_cuda_template() local