Home
last modified time | relevance | path

Searched defs:segment_ids_shape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops.cc1157 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local
1178 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local
1240 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dunsorted_segment.cc52 const tflite::RuntimeShape& segment_ids_shape, in UnsortedSegmentRef()
176 const tflite::RuntimeShape& segment_ids_shape, in EvalType()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunsorted_segment.cc144 const RuntimeShape& segment_ids_shape, in EvalType()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreference_ops.h995 const RuntimeShape& segment_ids_shape, in SegmentSum()
1015 const RuntimeShape& segment_ids_shape, in UnsortedSegmentRef()