Searched defs:data_contig (Results 1 – 2 of 2) sorted by relevance
186 const Tensor& data_contig, in _segment_reduce_cpu_lengths_backward_kernel1()322 const Tensor& data_contig, in _segment_reduce_cpu_lengths_backward_kernel()355 const Tensor& data_contig, in _segment_reduce_cpu_offsets_backward_kernel()409 const auto data_contig = data.contiguous(); in segment_reduce_kernel() local503 const auto data_contig = data.contiguous(); in _segment_reduce_backward_kernel() local
265 const Tensor& data_contig, in _segment_reduce_lengths_offsets_backward_cuda_kernel()367 const Tensor& data_contig, in _segment_reduce_lengths_backward_cuda_kernel()379 const Tensor& data_contig, in _segment_reduce_offsets_backward_cuda_kernel()