Searched defs:segment_offsets (Results 1 – 3 of 3) sorted by relevance
144 const Index* __restrict__ segment_offsets, // [nsegments + 1] in SegmentMeanNormalizeKernel()159 const Index* __restrict__ segment_offsets, // [nsegments + 1] in LaunchSegmentMeanNormalizeKernel()197 Tindex* __restrict__ segment_offsets) { // [nsegments + 1] in SegmentOffsetsKernel()227 Tindex* segment_offsets) { // [nsegments + 1] in LaunchSegmentOffsetsKernel()
216 OffsetIteratorT segment_offsets, // [num_segments + 1] in GpuSegmentedReduce()
141 const Tensor& segment_offsets = context->input(1); in Compute() local