Searched refs:UnsortedSegmentReductionShapeFn (Results 1 – 4 of 4) sorted by relevance
1321 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1331 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1341 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1351 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
127 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
209 Status UnsortedSegmentReductionShapeFn(InferenceContext* c);
2180 Status UnsortedSegmentReductionShapeFn(InferenceContext* c) { in UnsortedSegmentReductionShapeFn() function