| /external/tensorflow/tensorflow/lite/kernels/ |
| D | svdf.cc | 132 context->ResizeTensor(context, output, output_size_array)); in Prepare()
165 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_tensor, in Prepare()
180 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
195 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
212 context->ResizeTensor(context, float_weights_time, in Prepare()
226 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, zero_points, in Prepare()
241 context, context->ResizeTensor(context, row_sums, row_sums_size)); in Prepare()
255 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_temp, in Prepare()
|
| D | bidirectional_sequence_lstm.cc | 556 context->ResizeTensor(context, fw_output, fw_output_size)); in Prepare()
593 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_scratch_buffer, in Prepare()
620 context, context->ResizeTensor(context, bw_output, bw_output_size)); in Prepare()
655 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_scratch_buffer, in Prepare()
672 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
689 context, context->ResizeTensor(context, fw_activation_state_quantized, in Prepare()
705 context, context->ResizeTensor(context, bw_activation_state_quantized, in Prepare()
721 context->ResizeTensor(context, fw_cell_state_quantized, in Prepare()
737 context->ResizeTensor(context, bw_cell_state_quantized, in Prepare()
759 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()
[all …]
|
| D | basic_rnn.cc | 105 context->ResizeTensor(context, output, output_size_array)); in Prepare()
124 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
138 context->ResizeTensor(context, hidden_state_quantized, in Prepare()
151 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
166 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, accum_scratch, in Prepare()
179 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, zero_points, in Prepare()
194 context, context->ResizeTensor(context, row_sums, row_sums_size)); in Prepare()
|
| D | unidirectional_sequence_rnn.cc | 112 context->ResizeTensor(context, output, output_size_array)); in Prepare()
131 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
145 context->ResizeTensor(context, hidden_state_quantized, in Prepare()
158 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
173 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, accum_scratch, in Prepare()
186 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, zero_points, in Prepare()
201 context, context->ResizeTensor(context, row_sums, row_sums_size)); in Prepare()
|
| D | transpose_conv.cc | 103 TfLiteStatus ResizeTensor(TfLiteContext* context, in ResizeTensor() function
118 return context->ResizeTensor(context, tensor_to_resize, shape); in ResizeTensor()
187 return context->ResizeTensor(context, col2im, col2im_shape_array); in ResizeCol2ImTensor()
202 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, transposed_weights, in ResizeAndTransposeWeights()
326 TF_LITE_ENSURE_STATUS(ResizeTensor(context, output_shape, output)); in Prepare()
367 ResizeTensor(context, output_shape, scratch_buffer)); in Prepare()
600 TF_LITE_ENSURE_OK(context, ResizeTensor(context, output_shape, output)); in Eval()
639 ResizeTensor(context, output_shape, scratch_buffer)); in Eval()
658 ResizeTensor(context, output_shape, scratch_buffer)); in Eval()
675 ResizeTensor(context, output_shape, scratch_buffer)); in Eval()
|
| D | random_uniform.cc | 77 return context->ResizeTensor(context, output, in Prepare()
89 context->ResizeTensor(context, output, in EvalFloat()
134 context->ResizeTensor(context, output, in EvalInt()
|
| D | bidirectional_sequence_rnn.cc | 209 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
226 context, context->ResizeTensor(context, fw_hidden_state_quantized, in Prepare()
243 context, context->ResizeTensor(context, bw_hidden_state_quantized, in Prepare()
259 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
276 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, accum_scratch, in Prepare()
291 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, zero_points, in Prepare()
308 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_row_sums, in Prepare()
324 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_row_sums, in Prepare()
340 context->ResizeTensor(context, aux_input_quantized, in Prepare()
356 context, context->ResizeTensor(context, fw_output, fw_output_size_array)); in Prepare()
[all …]
|
| D | unidirectional_sequence_gru.cc | 192 context->ResizeTensor(context, output, output_size)); in Prepare()
199 context, context->ResizeTensor(context, output_state, in Prepare()
216 context, context->ResizeTensor(context, activation, activation_size)); in Prepare()
228 context->ResizeTensor(context, concat, concat_size)); in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 943 context->ResizeTensor(context, output, output_size)); in Prepare()
980 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in Prepare()
996 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
1012 context->ResizeTensor(context, output_state_quantized, in Prepare()
1027 context->ResizeTensor(context, cell_state_quantized, in Prepare()
1049 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()
1062 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_sf, in Prepare()
1078 context->ResizeTensor(context, prod_scaling_factors, in Prepare()
1098 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
1117 context, context->ResizeTensor(context, accum_scratch, accum_size)); in Prepare()
[all …]
|
| D | batch_matmul.cc | 121 TfLiteStatus stat = context->ResizeTensor(context, output, output_shape); in ResizeOutputTensor()
168 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in InitializeTemporaries()
197 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in InitializeTemporaries()
227 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in InitializeTemporaries()
241 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in InitializeTemporaries()
258 context, context->ResizeTensor(context, accum_scratch, accum_size)); in InitializeTemporaries()
270 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_offsets, in InitializeTemporaries()
284 context, context->ResizeTensor(context, row_sums, row_sums_size)); in InitializeTemporaries()
|
| D | hashtable_lookup.cc | 94 status = context->ResizeTensor(context, output, outputSize); in Prepare()
96 if (context->ResizeTensor(context, hits, hitSize) != kTfLiteOk) { in Prepare()
|
| D | lstm.cc | 105 return context->ResizeTensor(context, ledger, ledger_size); in make_ledger()
1368 context->ResizeTensor(context, output, output_size)); in Prepare()
1427 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in Prepare()
1446 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
1462 context->ResizeTensor(context, output_state_quantized, in Prepare()
1477 context->ResizeTensor(context, cell_state_quantized, in Prepare()
1498 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()
1511 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_sf, in Prepare()
1527 context->ResizeTensor(context, prod_scaling_factors, in Prepare()
1547 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
[all …]
|
| D | unique.cc | 62 return context->ResizeTensor(context, output_index_tensor, in Prepare()
101 context->ResizeTensor(context, unique_output, shape.release())); in EvalImpl()
|
| D | conv.cc | 473 auto output_status = context->ResizeTensor(context, output, output_size); in Prepare()
495 auto im2col_status = context->ResizeTensor(context, im2col, im2col_size); in Prepare()
517 context->ResizeTensor(context, hwcn_weights, hwcn_weights_size); in Prepare()
536 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
557 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
575 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, accum_scratch, in Prepare()
601 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_offsets, in Prepare()
617 context, context->ResizeTensor(context, row_sums, row_sums_size)); in Prepare()
|
| D | irfft2d.cc | 132 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, output, output_shape)); in ResizeOutputandTemporaryTensors()
142 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, fft_integer_working_area, in ResizeOutputandTemporaryTensors()
153 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, fft_double_working_area, in ResizeOutputandTemporaryTensors()
|
| D | conv3d.cc | 184 context->ResizeTensor(context, output, output_size)); in Prepare()
212 context->ResizeTensor(context, im2col, im2col_size)); in Prepare()
231 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, transposed_filter, in Prepare()
|
| D | rank.cc | 48 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, output, output_size)); in Prepare()
|
| D | if.cc | 130 context->ResizeTensor(context, output, output_size)); in Prepare()
194 context->ResizeTensor(context, output, output_size)); in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util_test.cc | 51 TfLiteStatus ResizeTensor(TfLiteContext* context, TfLiteTensor* tensor, in ResizeTensor() function
78 context.ResizeTensor = ResizeTensor; in TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/gradient/ |
| D | bcast_grad_args.cc | 104 context->ResizeTensor(context, output1, output1_shape)); in Invoke()
109 context->ResizeTensor(context, output2, output2_shape)); in Invoke()
192 context->ResizeTensor(context, output1, output1_shape)); in Invoke()
206 context->ResizeTensor(context, output2, output2_shape)); in Invoke()
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | text_encoder3s.cc | 66 context, context->ResizeTensor( in ResizeOutputTensors()
73 context, context->ResizeTensor( in ResizeOutputTensors()
82 context, context->ResizeTensor( in ResizeOutputTensors()
109 context->ResizeTensor(context, &output_lengths, in Prepare()
|
| D | dist_diversification.cc | 86 return context->ResizeTensor(context, &output_indices, in AllocateOutputIndexes()
104 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, &output_length, in Prepare()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_tokenizer_tflite.cc | 101 context->ResizeTensor( in Eval()
109 context, context->ResizeTensor( in Eval()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | ngrams.cc | 101 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_values, in Prepare()
115 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_row_splits, in Prepare()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_range_tflite.cc | 95 context->ResizeTensor(context, &rt_nested_splits_out, in EvalT()
117 context->ResizeTensor(context, &rt_dense_values_out, in EvalT()
|