Searched refs:output_segment_index (Results 1 – 1 of 1) sorted by relevance
182 const Index output_segment_index = segment_ids[input_segment_index]; in UnsortedSegmentCustomKernel() local183 if (output_segment_index < 0 || in UnsortedSegmentCustomKernel()184 output_segment_index >= output_outer_dim_size) { in UnsortedSegmentCustomKernel()188 output_segment_index * inner_dim_size + segment_offset; in UnsortedSegmentCustomKernel()