Searched refs:UnsortedSegmentReductionShapeFn (Results 1 – 4 of 4) sorted by relevance
206 Status UnsortedSegmentReductionShapeFn(InferenceContext* c);
2129 Status UnsortedSegmentReductionShapeFn(InferenceContext* c) { in UnsortedSegmentReductionShapeFn() function
1309 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1319 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1329 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);1339 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
127 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);