Home
last modified time | relevance | path

Searched refs:UnsortedSegmentReductionShapeFn (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops.cc1321 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
1331 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
1341 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
1351 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
H A Dstring_ops.cc127 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.h209 Status UnsortedSegmentReductionShapeFn(InferenceContext* c);
H A Dcommon_shape_fns.cc2180 Status UnsortedSegmentReductionShapeFn(InferenceContext* c) { in UnsortedSegmentReductionShapeFn() function