Searched refs:UnsortedSegmentReductionShapeFn (Results 1 – 4 of 4) sorted by relevance
203 Status UnsortedSegmentReductionShapeFn(InferenceContext* c);
2089 Status UnsortedSegmentReductionShapeFn(InferenceContext* c) { in UnsortedSegmentReductionShapeFn() function
1288 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1298 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1308 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1318 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
112 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);