Searched refs:s_num_segments (Results 1 – 1 of 1) sorted by relevance
1105 ShapeHandle s_num_segments = c->input(2); in UnsortedSegmentReductionShapeFn() local1106 TF_RETURN_IF_ERROR(c->WithRank(s_num_segments, 0, &s_num_segments)); in UnsortedSegmentReductionShapeFn()