Searched defs:data_size_axis (Results 1 – 2 of 2) sorted by relevance
100 const int64_t data_size_axis, in segment_reduce_forward_kernel()171 const int64_t data_size_axis, in segment_reduce_backward_kernel()305 auto data_size_axis = data_contig.size(axis); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local434 auto data_size_axis = data.size(axis); in _segment_reduce_lengths_offsets_cuda_kernel() local
49 auto data_size_axis = data.size(axis); in _segment_reduce_lengths_cpu_kernel1() local203 auto data_size_axis = data_contig.size(axis); in _segment_reduce_cpu_lengths_backward_kernel1() local