Searched refs:rrelu_with_noise_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
23 __global__ void rrelu_with_noise_cuda_kernel( in rrelu_with_noise_cuda_kernel() function109 rrelu_with_noise_cuda_kernel<scalar_t, 2><<<grid, block, 0, stream>>>( in _rrelu_with_noise_cuda_train()123 rrelu_with_noise_cuda_kernel<scalar_t, 4><<<grid, block, 0, stream>>>( in _rrelu_with_noise_cuda_train()