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