Searched refs:tensor_kernel_scan_outer_dim (Results 1 – 1 of 1) sorted by relevance
272 __global__ void tensor_kernel_scan_outer_dim(scalar_t *tgt_, const scalar_t *src_, in tensor_kernel_scan_outer_dim() function410 tensor_kernel_scan_outer_dim<scalar_t><<<grid, threads, 0, at::cuda::getCurrentCUDAStream()>>>( in scan_outer_dim()