Searched refs:UnsortedSegmentRef (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A D | unsorted_segment.cc | 50 void UnsortedSegmentRef(const tflite::RuntimeShape& input_shape, in UnsortedSegmentRef() function
182 unsorted_segment::UnsortedSegmentRef<T, SegmenProd>( in EvalType()
187 unsorted_segment::UnsortedSegmentRef<T, SegmenMax>( in EvalType()
192 unsorted_segment::UnsortedSegmentRef<T, SegmenSum>( in EvalType()
197 unsorted_segment::UnsortedSegmentRef<T, SegmenMin>( in EvalType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unsorted_segment.cc | 150 reference_ops::UnsortedSegmentRef<T, SegmenProd>( in EvalType()
155 reference_ops::UnsortedSegmentRef<T, SegmenMax>( in EvalType()
160 reference_ops::UnsortedSegmentRef<T, SegmenSum>( in EvalType()
165 reference_ops::UnsortedSegmentRef<T, SegmenMin>( in EvalType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 1013 inline void UnsortedSegmentRef(const RuntimeShape& input_shape, in UnsortedSegmentRef() function
|