Searched refs:kInputDataTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | segment_sum.cc | 29 static const int kInputDataTensor = 0; variable
69 GetInputSafe(context, node, kInputDataTensor, &data)); in Prepare()
91 GetInputSafe(context, node, kInputDataTensor, &data)); in Eval()
|
| H A D | unsorted_segment.cc | 38 static const int kInputDataTensor = 0; variable
93 GetInputSafe(context, node, kInputDataTensor, &data)); in Prepare()
181 GetInputSafe(context, node, kInputDataTensor, &data)); in EvalGeneric()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A D | unsorted_segment.cc | 39 static const int kInputDataTensor = 0; variable
125 tflite::GetInputSafe(context, node, kInputDataTensor, &data)); in Prepare()
213 tflite::GetInputSafe(context, node, kInputDataTensor, &data)); in EvalGeneric()
|