Home
last modified time | relevance | path

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

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