Searched refs:segment_ids_data (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A D | unsorted_segment.cc | 53 const int32_t* segment_ids_data, in UnsortedSegmentRef() argument
65 int output_index = segment_ids_data[i]; in UnsortedSegmentRef()
177 const int32_t* segment_ids_data, in EvalType() argument
183 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
188 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
193 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
198 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unsorted_segment.cc | 145 const int32_t* segment_ids_data, in EvalType() argument
151 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
156 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
161 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
166 input_shape, input_data, segment_ids_shape, segment_ids_data, in EvalType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2020-004.md | 13 int output_index = segment_ids_data[i];
21 Users having access to `segment_ids_data` can alter `output_index` and then
|
| H A D | tfsa-2020-002.md | 26 int output_index = segment_ids_data[i];
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 996 const int32_t* segment_ids_data, in SegmentSum() argument
1004 int output_index = segment_ids_data[i]; in SegmentSum()
1016 const int32_t* segment_ids_data, in UnsortedSegmentRef() argument
1028 int output_index = segment_ids_data[i]; in UnsortedSegmentRef()
|