Searched defs:output_stride_axis (Results 1 – 2 of 2) sorted by relevance
101 const int64_t output_stride_axis, in segment_reduce_forward_kernel()172 const int64_t output_stride_axis, in segment_reduce_backward_kernel()306 auto output_stride_axis = output_contig.stride(axis); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local435 auto output_stride_axis = output.stride(axis); in _segment_reduce_lengths_offsets_cuda_kernel() local
50 auto output_stride_axis = output.stride(axis); in _segment_reduce_lengths_cpu_kernel1() local204 auto output_stride_axis = output_contig.stride(axis); in _segment_reduce_cpu_lengths_backward_kernel1() local