Searched defs:cumsum_ptr (Results 1 – 3 of 3) sorted by relevance
16 const int64_t* cumsum_ptr, in compute_cuda_kernel()39 const int64_t* cumsum_ptr, in compute_cuda()
42 const int64_t* cumsum_ptr = cumsum.const_data_ptr<int64_t>(); in repeat_interleave_common() local
20 const int64_t* cumsum_ptr, in compute_cpu()