Searched refs:SparseSegmentReductionWithNumSegmentsShapeFn (Results 1 – 1 of 1) sorted by relevance
1063 Status SparseSegmentReductionWithNumSegmentsShapeFn(InferenceContext* c) { in SparseSegmentReductionWithNumSegmentsShapeFn() function1232 .SetShapeFn(SparseSegmentReductionWithNumSegmentsShapeFn);1252 .SetShapeFn(SparseSegmentReductionWithNumSegmentsShapeFn);1282 .SetShapeFn(SparseSegmentReductionWithNumSegmentsShapeFn);