Home
last modified time | relevance | path

Searched refs:SetTensorToDynamic (Results 1 – 25 of 58) sorted by relevance

123

/external/libtextclassifier/native/utils/tflite/
Dtext_encoder3s.cc132 tflite::SetTensorToDynamic(&output_encoded); in Prepare()
133 tflite::SetTensorToDynamic(&output_positions); in Prepare()
137 tflite::SetTensorToDynamic(&output_attr); in Prepare()
Dtext_encoder.cc198 tflite::SetTensorToDynamic(&output_encoded); in Prepare()
199 tflite::SetTensorToDynamic(&output_positions); in Prepare()
203 tflite::SetTensorToDynamic(&output_attr); in Prepare()
Dtoken_encoder.cc108 tflite::SetTensorToDynamic(&output_positions); in Prepare()
112 tflite::SetTensorToDynamic(&output_attr); in Prepare()
Ddist_diversification.cc100 tflite::SetTensorToDynamic(&output_indices); in Prepare()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_tokenizer_tflite.cc60 SetTensorToDynamic(&output_values); in Prepare()
64 SetTensorToDynamic(&output_splits); in Prepare()
Dsentencepiece_detokenizer_tflite.cc48 SetTensorToDynamic(&output_values); in Prepare()
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc155 SetTensorToDynamic(output_selected_indices); in Prepare()
156 SetTensorToDynamic(output_selected_scores); in Prepare()
175 SetTensorToDynamic(output_selected_indices); in Prepare()
Dirfft2d.cc219 SetTensorToDynamic(fft_integer_working_area); in Prepare()
220 SetTensorToDynamic(fft_double_working_area); in Prepare()
221 SetTensorToDynamic(output); in Prepare()
Dif.cc95 SetTensorToDynamic(subgraph_input); in Prepare()
125 SetTensorToDynamic(output); in Prepare()
Drfft2d.cc219 SetTensorToDynamic(fft_integer_working_area); in Prepare()
220 SetTensorToDynamic(fft_double_working_area); in Prepare()
221 SetTensorToDynamic(output); in Prepare()
Dread_variable.cc52 SetTensorToDynamic(output); in Prepare()
Dwhile.cc293 SetTensorToDynamic(body_input); in Prepare_impl()
325 SetTensorToDynamic(output); in Prepare_impl()
347 SetTensorToDynamic(output); in Prepare()
Drandom_standard_normal_custom.cc67 SetTensorToDynamic(output); in Prepare()
Dtopk_v2.cc227 SetTensorToDynamic(output_indexes); in Prepare()
228 SetTensorToDynamic(output_values); in Prepare()
Drandom_ops.cc189 SetTensorToDynamic(output); in Prepare()
218 SetTensorToDynamic(output); in PrepareMultinomial()
Dconv3d_transpose.cc197 SetTensorToDynamic(output); in Prepare()
199 SetTensorToDynamic(col2im); in Prepare()
Dbroadcast_to.cc108 SetTensorToDynamic(op_context.output); in Prepare()
Dsegment_sum.cc81 SetTensorToDynamic(output); in Prepare()
Dexpand_dims.cc98 SetTensorToDynamic(output); in Prepare()
Dtranspose_conv.cc322 SetTensorToDynamic(output); in Prepare()
324 SetTensorToDynamic(col2im); in Prepare()
343 SetTensorToDynamic(transposed_weights); in Prepare()
365 SetTensorToDynamic(scratch_buffer); in Prepare()
Dresize_bilinear.cc84 SetTensorToDynamic(output); in Prepare()
Dfill.cc106 SetTensorToDynamic(output); in Prepare()
/external/tensorflow/tensorflow/lite/kernels/ctc/
Dctc_beam_search_decoder.cc87 SetTensorToDynamic(indices); in Prepare()
91 SetTensorToDynamic(values); in Prepare()
95 SetTensorToDynamic(output_shape); in Prepare()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc159 SetTensorToDynamic(&nested_splits); in Prepare()
162 SetTensorToDynamic(&dense_values); in Prepare()
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc64 SetTensorToDynamic(output1); in Prepare()
65 SetTensorToDynamic(output2); in Prepare()

123