Searched refs:s_data_suffix (Results 1 – 1 of 1) sorted by relevance
1121 ShapeHandle s_data_suffix; in UnsortedSegmentReductionShapeFn() local1123 c->Subshape(s_data, c->Rank(s_segment_ids), &s_data_suffix)); in UnsortedSegmentReductionShapeFn()1125 c->Concatenate(c->Vector(num_segments_dim), s_data_suffix, &out)); in UnsortedSegmentReductionShapeFn()