Searched refs:s_segment_ids (Results 1 – 1 of 1) sorted by relevance
2091 ShapeHandle s_segment_ids = c->input(1); in UnsortedSegmentReductionShapeFn() local2099 if (c->RankKnown(s_segment_ids)) { in UnsortedSegmentReductionShapeFn()2101 c->MergePrefix(s_data, s_segment_ids, &s_data, &s_segment_ids)); in UnsortedSegmentReductionShapeFn()2110 c->Subshape(s_data, c->Rank(s_segment_ids), &s_data_suffix)); in UnsortedSegmentReductionShapeFn()