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