Searched defs:data_stride_axis (Results 1 – 2 of 2) sorted by relevance
99 const int64_t data_stride_axis, in segment_reduce_forward_kernel()170 const int64_t data_stride_axis, in segment_reduce_backward_kernel()304 auto data_stride_axis = data_contig.stride(axis); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local433 auto data_stride_axis = data.stride(axis); in _segment_reduce_lengths_offsets_cuda_kernel() local
48 auto data_stride_axis = data.stride(axis); in _segment_reduce_lengths_cpu_kernel1() local202 auto data_stride_axis = data_contig.stride(axis); in _segment_reduce_cpu_lengths_backward_kernel1() local