Searched refs:num_segments_dim (Results 1 – 1 of 1) sorted by relevance
2195 DimensionHandle num_segments_dim; in UnsortedSegmentReductionShapeFn() local2196 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &num_segments_dim)); in UnsortedSegmentReductionShapeFn()2203 c->Concatenate(c->Vector(num_segments_dim), s_data_suffix, &out)); in UnsortedSegmentReductionShapeFn()