| /external/tensorflow/tensorflow/lite/kernels/ |
| D | svdf.cc | 131 context->ResizeTensor(context, output, output_size_array)); in Prepare()
164 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_tensor, in Prepare()
179 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
194 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
211 context->ResizeTensor(context, float_weights_time, in Prepare()
225 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, zero_points, in Prepare()
240 context, context->ResizeTensor(context, row_sums, row_sums_size)); in Prepare()
254 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_temp, in Prepare()
|
| 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 | 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 | 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 | random_uniform.cc | 74 return context->ResizeTensor(context, output, in Prepare()
86 context->ResizeTensor(context, output, in EvalFloat()
131 context->ResizeTensor(context, output, in EvalInt()
|
| 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()
596 TF_LITE_ENSURE_OK(context, ResizeTensor(context, output_shape, output)); in Eval()
635 ResizeTensor(context, output_shape, scratch_buffer)); in Eval()
654 ResizeTensor(context, output_shape, scratch_buffer)); in Eval()
671 ResizeTensor(context, output_shape, scratch_buffer)); in Eval()
|
| 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 | 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 | unidirectional_sequence_lstm.cc | 934 context->ResizeTensor(context, output, output_size)); in Prepare()
971 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in Prepare()
987 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
1003 context->ResizeTensor(context, output_state_quantized, in Prepare()
1018 context->ResizeTensor(context, cell_state_quantized, in Prepare()
1040 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()
1053 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_sf, in Prepare()
1069 context->ResizeTensor(context, prod_scaling_factors, in Prepare()
1089 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
1108 context, context->ResizeTensor(context, accum_scratch, accum_size)); in Prepare()
[all …]
|
| D | hashtable_lookup.cc | 94 status = context->ResizeTensor(context, output, outputSize); in Prepare()
96 if (context->ResizeTensor(context, hits, hitSize) != kTfLiteOk) { in Prepare()
|
| D | unique.cc | 62 return context->ResizeTensor(context, output_index_tensor, in Prepare()
101 context->ResizeTensor(context, unique_output, shape.release())); in EvalImpl()
|
| 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 | rank.cc | 48 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, output, output_size)); in Prepare()
|
| D | conv.cc | 449 auto output_status = context->ResizeTensor(context, output, output_size); in Prepare()
471 auto im2col_status = context->ResizeTensor(context, im2col, im2col_size); in Prepare()
493 context->ResizeTensor(context, hwcn_weights, hwcn_weights_size); in Prepare()
512 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
532 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
550 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, accum_scratch, in Prepare()
575 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_offsets, in Prepare()
591 context, context->ResizeTensor(context, row_sums, row_sums_size)); in Prepare()
|
| D | round.cc | 43 return context->ResizeTensor(context, output, output_size); in Prepare()
|
| D | fully_connected.cc | 64 return context->ResizeTensor(context, ledger, ledger_size); in CreateLedgerTensor()
279 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in PrepareImpl()
293 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in PrepareImpl()
310 context, context->ResizeTensor(context, accum_scratch, accum_size)); in PrepareImpl()
322 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_offsets, in PrepareImpl()
336 context, context->ResizeTensor(context, row_sums, row_sums_size)); in PrepareImpl()
368 context->ResizeTensor(context, output, output_size_array)); in PrepareImpl()
|
| D | if.cc | 130 context->ResizeTensor(context, output, output_size)); in Prepare()
194 context->ResizeTensor(context, output, output_size)); in Eval()
|
| D | zeros_like.cc | 42 return context->ResizeTensor(context, output, in Prepare()
|
| D | ceil.cc | 42 return context->ResizeTensor(context, output, output_size); in Prepare()
|
| /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/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 184 context->ResizeTensor(context, output, output_size)); in Prepare()
191 context, context->ResizeTensor(context, output_state, in Prepare()
208 context, context->ResizeTensor(context, activation, activation_size)); in Prepare()
220 context->ResizeTensor(context, concat, concat_size)); in Prepare()
|
| /external/libtextclassifier/native/utils/tflite/ |
| 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/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()
|
| /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()
|