Searched refs:tensor_kernel_scan_outer_dim_with_indices (Results 1 – 1 of 1) sorted by relevance
155 __global__ void tensor_kernel_scan_outer_dim_with_indices(const scalar_t *self_, scalar_t *values_,… in tensor_kernel_scan_outer_dim_with_indices() function210 …tensor_kernel_scan_outer_dim_with_indices<scalar_t><<<grid, threads, 0, at::cuda::getCurrentCUDASt… in scan_outer_dim_with_indices()