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