Searched refs:input_segment_index (Results 1 – 1 of 1) sorted by relevance
180 const int64_t input_segment_index = input_index / inner_dim_size; in UnsortedSegmentCustomKernel() local182 const Index output_segment_index = segment_ids[input_segment_index]; in UnsortedSegmentCustomKernel()