Searched refs:num_segments_shape (Results 1 – 1 of 1) sorted by relevance
1073 ShapeHandle num_segments_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local1074 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &num_segments_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()