| /aosp_15_r20/external/libtextclassifier/native/utils/tflite/ |
| H A D | text_encoder3s.cc | 132 tflite::SetTensorToDynamic(&output_encoded); in Prepare()
133 tflite::SetTensorToDynamic(&output_positions); in Prepare()
137 tflite::SetTensorToDynamic(&output_attr); in Prepare()
|
| H A D | text_encoder.cc | 198 tflite::SetTensorToDynamic(&output_encoded); in Prepare()
199 tflite::SetTensorToDynamic(&output_positions); in Prepare()
203 tflite::SetTensorToDynamic(&output_attr); in Prepare()
|
| H A D | token_encoder.cc | 108 tflite::SetTensorToDynamic(&output_positions); in Prepare()
112 tflite::SetTensorToDynamic(&output_attr); in Prepare()
|
| H A D | dist_diversification.cc | 100 tflite::SetTensorToDynamic(&output_indices); in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_beam_search_decoder.cc | 87 SetTensorToDynamic(indices); in Prepare()
91 SetTensorToDynamic(values); in Prepare()
95 SetTensorToDynamic(output_shape); in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | non_max_suppression.cc | 155 SetTensorToDynamic(output_selected_indices); in Prepare()
156 SetTensorToDynamic(output_selected_scores); in Prepare()
175 SetTensorToDynamic(output_selected_indices); in Prepare()
|
| H A D | irfft2d.cc | 219 SetTensorToDynamic(fft_integer_working_area); in Prepare()
220 SetTensorToDynamic(fft_double_working_area); in Prepare()
221 SetTensorToDynamic(output); in Prepare()
|
| H A D | rfft2d.cc | 219 SetTensorToDynamic(fft_integer_working_area); in Prepare()
220 SetTensorToDynamic(fft_double_working_area); in Prepare()
221 SetTensorToDynamic(output); in Prepare()
|
| H A D | while.cc | 293 SetTensorToDynamic(body_input); in Prepare_impl()
325 SetTensorToDynamic(output); in Prepare_impl()
347 SetTensorToDynamic(output); in Prepare()
|
| H A D | if.cc | 95 SetTensorToDynamic(subgraph_input); in Prepare()
125 SetTensorToDynamic(output); in Prepare()
|
| H A D | transpose_conv.cc | 322 SetTensorToDynamic(output); in Prepare()
324 SetTensorToDynamic(col2im); in Prepare()
343 SetTensorToDynamic(transposed_weights); in Prepare()
365 SetTensorToDynamic(scratch_buffer); in Prepare()
|
| H A D | reduce.cc | 257 SetTensorToDynamic(normalized_dims); in PrepareSimple()
265 SetTensorToDynamic(op_context.output); in PrepareSimple()
266 SetTensorToDynamic(resolved_axis); in PrepareSimple()
310 SetTensorToDynamic(temp_sum); in PrepareMeanOrSum()
342 SetTensorToDynamic(temp_prod); in PrepareProd()
|
| H A D | random_ops.cc | 189 SetTensorToDynamic(output); in Prepare()
218 SetTensorToDynamic(output); in PrepareMultinomial()
|
| H A D | topk_v2.cc | 227 SetTensorToDynamic(output_indexes); in Prepare()
228 SetTensorToDynamic(output_values); in Prepare()
|
| H A D | conv3d_transpose.cc | 197 SetTensorToDynamic(output); in Prepare()
199 SetTensorToDynamic(col2im); in Prepare()
|
| H A D | read_variable.cc | 52 SetTensorToDynamic(output); in Prepare()
|
| H A D | random_standard_normal_custom.cc | 67 SetTensorToDynamic(output); in Prepare()
|
| H A D | segment_sum.cc | 81 SetTensorToDynamic(output); in Prepare()
|
| H A D | broadcast_to.cc | 108 SetTensorToDynamic(op_context.output); in Prepare()
|
| H A D | expand_dims.cc | 98 SetTensorToDynamic(output); in Prepare()
|
| H A D | split.cc | 46 SetTensorToDynamic(tensor); in UseDynamicOutputTensors()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| H A D | sentencepiece_tokenizer_tflite.cc | 60 SetTensorToDynamic(&output_values); in Prepare()
64 SetTensorToDynamic(&output_splits); in Prepare()
|
| H A D | sentencepiece_detokenizer_tflite.cc | 48 SetTensorToDynamic(&output_values); in Prepare()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_range_tflite.cc | 159 SetTensorToDynamic(&nested_splits); in Prepare()
162 SetTensorToDynamic(&dense_values); in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/gradient/ |
| H A D | bcast_grad_args.cc | 64 SetTensorToDynamic(output1); in Prepare()
65 SetTensorToDynamic(output2); in Prepare()
|