Home
last modified time | relevance | path

Searched refs:nll_loss_backward_reduce_cuda_kernel_1d (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLoss.cu429 __global__ void nll_loss_backward_reduce_cuda_kernel_1d( in nll_loss_backward_reduce_cuda_kernel_1d() function
545 nll_loss_backward_reduce_cuda_kernel_1d<scalar_t, index_t> in nll_loss_backward_out_cuda_template()