Searched refs:kInputNumSegmentsTensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unsorted_segment.cc | 40 static const int kInputNumSegmentsTensor = 2; variable
100 GetInputSafe(context, node, kInputNumSegmentsTensor, &num_segments)); in Prepare()
188 GetInputSafe(context, node, kInputNumSegmentsTensor, &num_segments)); in EvalGeneric()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A D | unsorted_segment.cc | 41 static const int kInputNumSegmentsTensor = 2; variable
132 tflite::GetInputSafe(context, node, kInputNumSegmentsTensor, &num_segments)); in Prepare()
220 tflite::GetInputSafe(context, node, kInputNumSegmentsTensor, &num_segments)); in EvalGeneric()
|