Home
last modified time | relevance | path

Searched defs:segment_flat_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dunsorted_segment.cc60 int segment_flat_size = 1; in UnsortedSegmentRef() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreference_ops.h998 const int segment_flat_size = in SegmentSum() local
1023 int segment_flat_size = 1; in UnsortedSegmentRef() local