| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | where.cc | 34 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
71 return ResizeOutputTensor<T>(context, cond_tensor, output); in PrepareOutput()
120 context, ResizeOutputTensor<bool>(context, cond_tensor, output)); in Eval()
124 context, ResizeOutputTensor<float>(context, cond_tensor, output)); in Eval()
128 context, ResizeOutputTensor<int64_t>(context, cond_tensor, output)); in Eval()
132 context, ResizeOutputTensor<int32_t>(context, cond_tensor, output)); in Eval()
136 context, ResizeOutputTensor<int8_t>(context, cond_tensor, output)); in Eval()
140 context, ResizeOutputTensor<uint8_t>(context, cond_tensor, output)); in Eval()
143 TF_LITE_ENSURE_OK(context, ResizeOutputTensor<uint32_t>( in Eval()
|
| H A D | segment_sum.cc | 33 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
85 return ResizeOutputTensor(context, data, segment_ids, output); in Prepare()
101 ResizeOutputTensor(context, data, segment_ids, output)); in Eval()
|
| H A D | broadcast_to.cc | 47 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
105 return ResizeOutputTensor(context, &op_context); in Prepare()
115 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in Eval()
|
| H A D | resize_bilinear.cc | 46 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
97 return ResizeOutputTensor(context, input, size, output); in Prepare()
115 ResizeOutputTensor(context, input, size, output)); in Eval()
|
| H A D | transpose.cc | 48 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
87 return ResizeOutputTensor(context, &op_context); in Prepare()
96 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in Eval()
|
| H A D | batch_to_space_nd.cc | 56 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
111 return ResizeOutputTensor(context, &op_context); in Prepare()
120 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in Eval()
|
| H A D | one_hot.cc | 111 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
165 return ResizeOutputTensor(context, op_context); in Prepare()
172 ResizeOutputTensor(context, op_context); in Eval()
|
| H A D | space_to_batch_nd.cc | 57 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
112 return ResizeOutputTensor(context, &op_context); in Prepare()
121 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in Eval()
|
| H A D | resize_nearest_neighbor.cc | 46 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
84 return ResizeOutputTensor(context, input, size, output); in Prepare()
102 ResizeOutputTensor(context, input, size, output)); in Eval()
|
| H A D | scatter_nd.cc | 36 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
116 return ResizeOutputTensor<int32_t>(context, shape, output); in Prepare()
148 ResizeOutputTensor<IndicesT>(context, shape, output)); in EvalScatterNd()
|
| H A D | unsorted_segment.cc | 43 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
114 return ResizeOutputTensor(context, data, segment_ids, num_segments, output); in Prepare()
194 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, data, segment_ids, in EvalGeneric()
|
| H A D | pad.cc | 75 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
133 return ResizeOutputTensor(context, &op_context); in Prepare()
185 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in Eval()
|
| H A D | strided_slice.cc | 186 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
263 return ResizeOutputTensor(context, &op_context); in Prepare()
271 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in Eval()
|
| H A D | reduce.cc | 101 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, OpContext* op_context) { in ResizeOutputTensor() function
272 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in PrepareSimple()
506 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in EvalMean()
714 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, op_context)); in EvalType()
877 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in EvalSum()
936 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, op_context)); in EvalQuantizedProd()
|
| H A D | batch_matmul.cc | 99 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
386 ResizeOutputTensor(context, extended_lhs_shape, extended_rhs_shape, adj_x, in Prepare()
|
| /aosp_15_r20/external/libtextclassifier/native/utils/tflite/ |
| H A D | text_encoder.cc | 141 ResizeOutputTensor(max_output_length, in ResizeOutputTensors()
147 ResizeOutputTensor( in ResizeOutputTensors()
155 ResizeOutputTensor( in ResizeOutputTensors()
|
| H A D | token_encoder.cc | 59 ResizeOutputTensor( in ResizeOutputTensors()
67 ResizeOutputTensor( in ResizeOutputTensors()
|
| H A D | encoder_common.h | 52 TfLiteStatus ResizeOutputTensor(const int max_output_length,
|
| H A D | encoder_common.cc | 103 TfLiteStatus ResizeOutputTensor(const int max_output_length, in ResizeOutputTensor() function
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A D | unsorted_segment.cc | 75 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, in ResizeOutputTensor() function
146 return ResizeOutputTensor(context, data, segment_ids, num_segments, output); in Prepare()
226 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, data, segment_ids, in EvalGeneric()
|