Home
last modified time | relevance | path

Searched refs:TfLiteIntArrayCreate (Results 1 – 25 of 144) sorted by relevance

123456

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dkernel_util_test.cc76 tensor->dims = TfLiteIntArrayCreate(dims.size()); in SetShape()
335 TfLiteIntArray* dims0 = TfLiteIntArrayCreate(0); in TEST_F()
339 TfLiteIntArray* dims1 = TfLiteIntArrayCreate(1); in TEST_F()
344 TfLiteIntArray* dims2 = TfLiteIntArrayCreate(2); in TEST_F()
350 TfLiteIntArray* dims3 = TfLiteIntArrayCreate(3); in TEST_F()
363 input.dims = TfLiteIntArrayCreate(1); in TEST_F()
372 input_params->zero_point = TfLiteIntArrayCreate(1); in TEST_F()
380 filter.dims = TfLiteIntArrayCreate(4); in TEST_F()
394 filter_params->zero_point = TfLiteIntArrayCreate(3); in TEST_F()
405 bias.dims = TfLiteIntArrayCreate(4); in TEST_F()
[all …]
H A Dsvdf.cc128 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
141 node->temporaries = TfLiteIntArrayCreate(6); in Prepare()
143 node->temporaries = TfLiteIntArrayCreate(2); in Prepare()
145 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
149 TfLiteIntArray* scratch_size_array = TfLiteIntArrayCreate(2); in Prepare()
193 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
225 TfLiteIntArray* zero_points_size = TfLiteIntArrayCreate(1); in Prepare()
240 TfLiteIntArray* row_sums_size = TfLiteIntArrayCreate(1); in Prepare()
248 TfLiteIntArray* output_temp_size_array = TfLiteIntArrayCreate(2); in Prepare()
H A Dbidirectional_sequence_lstm.cc550 TfLiteIntArray* fw_output_size = TfLiteIntArrayCreate(3); in Prepare()
563 node->temporaries = TfLiteIntArrayCreate( in Prepare()
566 node->temporaries = TfLiteIntArrayCreate(2); // the two scratch buffers. in Prepare()
584 TfLiteIntArray* fw_scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
617 TfLiteIntArray* bw_output_size = TfLiteIntArrayCreate(3); in Prepare()
648 TfLiteIntArray* bw_scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
760 TfLiteIntArray* input_sf_size = TfLiteIntArrayCreate(1); in Prepare()
774 TfLiteIntArray* aux_input_sf_size = TfLiteIntArrayCreate(1); in Prepare()
788 TfLiteIntArray* output_state_sf_size = TfLiteIntArrayCreate(1); in Prepare()
803 TfLiteIntArray* prod_scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
[all …]
H A Dbasic_rnn.cc101 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
115 node->temporaries = TfLiteIntArrayCreate(6); in Prepare()
149 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
163 TfLiteIntArray* accum_scratch_size = TfLiteIntArrayCreate(2); in Prepare()
177 TfLiteIntArray* zero_points_size = TfLiteIntArrayCreate(1); in Prepare()
191 TfLiteIntArray* row_sums_size = TfLiteIntArrayCreate(2); in Prepare()
H A Dunidirectional_sequence_rnn.cc107 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(3); in Prepare()
122 node->temporaries = TfLiteIntArrayCreate(6); in Prepare()
156 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
170 TfLiteIntArray* accum_scratch_size = TfLiteIntArrayCreate(2); in Prepare()
184 TfLiteIntArray* zero_points_size = TfLiteIntArrayCreate(1); in Prepare()
197 TfLiteIntArray* row_sums_size = TfLiteIntArrayCreate(2); in Prepare()
H A Dbatch_matmul.cc104 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(output_rank); in ResizeOutputTensor()
139 node->temporaries = TfLiteIntArrayCreate(kNumTempTensorsForAdjoints + in InitializeTemporaries()
142 node->temporaries = TfLiteIntArrayCreate(kNumTempTensorsForAdjoints); in InitializeTemporaries()
160 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(lhs_rank); in InitializeTemporaries()
186 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(rhs_rank); in InitializeTemporaries()
242 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
257 TfLiteIntArray* accum_size = TfLiteIntArrayCreate(2); in InitializeTemporaries()
271 TfLiteIntArray* input_offsets_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
284 TfLiteIntArray* row_sums_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
H A Dunidirectional_sequence_lstm.cc953 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
955 node->temporaries = TfLiteIntArrayCreate(6); in Prepare()
957 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
972 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
1049 TfLiteIntArray* input_sf_size = TfLiteIntArrayCreate(1); in Prepare()
1063 TfLiteIntArray* output_state_sf_size = TfLiteIntArrayCreate(1); in Prepare()
1078 TfLiteIntArray* prod_scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
1098 TfLiteIntArray* recovered_cell_weights_size = TfLiteIntArrayCreate(1); in Prepare()
1116 TfLiteIntArray* accum_size = TfLiteIntArrayCreate(2); in Prepare()
1130 TfLiteIntArray* input_zp_size = TfLiteIntArrayCreate(1); in Prepare()
[all …]
H A Dlstm.cc103 TfLiteIntArray* ledger_size = TfLiteIntArrayCreate(1); in make_ledger()
1365 TfLiteIntArray* output_size = TfLiteIntArrayCreate(2); in Prepare()
1390 TfLiteIntArrayCreate(kNumHybridTemporaryTensors + kLedgersToAdd); in Prepare()
1392 node->temporaries = TfLiteIntArrayCreate(kNumHybridTemporaryTensors); in Prepare()
1396 node->temporaries = TfLiteIntArrayCreate(6); in Prepare()
1398 node->temporaries = TfLiteIntArrayCreate(8); in Prepare()
1401 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
1419 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
1498 TfLiteIntArray* input_sf_size = TfLiteIntArrayCreate(1); in Prepare()
1512 TfLiteIntArray* output_state_sf_size = TfLiteIntArrayCreate(1); in Prepare()
[all …]
H A Dbidirectional_sequence_rnn.cc194 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
197 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors - 1); in Prepare()
257 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
273 TfLiteIntArray* accum_scratch_size = TfLiteIntArrayCreate(2); in Prepare()
289 TfLiteIntArray* zero_points_size = TfLiteIntArrayCreate(1); in Prepare()
306 TfLiteIntArray* fw_row_sums_size = TfLiteIntArrayCreate(2); in Prepare()
323 TfLiteIntArray* bw_row_sums_size = TfLiteIntArrayCreate(2); in Prepare()
352 TfLiteIntArray* fw_output_size_array = TfLiteIntArrayCreate(3); in Prepare()
363 TfLiteIntArray* bw_output_size_array = TfLiteIntArrayCreate(3); in Prepare()
H A Dunidirectional_sequence_gru.cc187 TfLiteIntArray* output_size = TfLiteIntArrayCreate(3); in Prepare()
203 node->temporaries = TfLiteIntArrayCreate(kTemporaryNum); in Prepare()
212 TfLiteIntArray* activation_size = TfLiteIntArrayCreate(2); in Prepare()
224 TfLiteIntArray* concat_size = TfLiteIntArrayCreate(2); in Prepare()
H A Dconv3d.cc117 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in AllocateTemporaryTensorsIfRequired()
177 TfLiteIntArray* output_size = TfLiteIntArrayCreate(5); in Prepare()
197 TfLiteIntArray* im2col_size = TfLiteIntArrayCreate(5); in Prepare()
216 TfLiteIntArray* transposed_filter_size = TfLiteIntArrayCreate(5); in Prepare()
H A Dreduce.cc87 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis()
95 TfLiteIntArray* size = TfLiteIntArrayCreate(1); in ResizeTempAccum()
107 TfLiteIntArrayCreate(0)); in ResizeOutputTensor()
111 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims); in ResizeOutputTensor()
149 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis); in ResizeOutputTensor()
171 TfLiteIntArray* dims_size = TfLiteIntArrayCreate(1); in ResizeTempDims()
182 node->temporaries = TfLiteIntArrayCreate(4); in InitializeTemporaries()
189 TfLiteIntArray* index_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
H A Dconv.cc328 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in AllocateTemporaryTensorsIfRequired()
489 TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); in Prepare()
501 TfLiteIntArray* im2col_size = TfLiteIntArrayCreate(4); in Prepare()
522 TfLiteIntArray* hwcn_weights_size = TfLiteIntArrayCreate(2); in Prepare()
578 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
595 TfLiteIntArray* accum_scratch_size = TfLiteIntArrayCreate(2); in Prepare()
622 TfLiteIntArray* input_offsets_size = TfLiteIntArrayCreate(1); in Prepare()
638 TfLiteIntArray* row_sums_size = TfLiteIntArrayCreate(1); in Prepare()
H A Dfully_connected.cc62 TfLiteIntArray* ledger_size = TfLiteIntArrayCreate(1); in CreateLedgerTensor()
240 output_size_array = TfLiteIntArrayCreate(2); in PrepareImpl()
343 node->temporaries = TfLiteIntArrayCreate(6); in PrepareImpl()
345 node->temporaries = TfLiteIntArrayCreate(5); in PrepareImpl()
368 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in PrepareImpl()
383 TfLiteIntArray* accum_size = TfLiteIntArrayCreate(2); in PrepareImpl()
397 TfLiteIntArray* input_offsets_size = TfLiteIntArrayCreate(1); in PrepareImpl()
410 TfLiteIntArray* row_sums_size = TfLiteIntArrayCreate(1); in PrepareImpl()
440 output_size_array = TfLiteIntArrayCreate(2); in PrepareImpl()
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon_test.cc31 TfLiteIntArray* a = TfLiteIntArrayCreate(0); in TEST()
32 TfLiteIntArray* b = TfLiteIntArrayCreate(3); in TEST()
38 TfLiteIntArray* a = TfLiteIntArrayCreate(2); in TEST()
51 TfLiteIntArray* a = TfLiteIntArrayCreate(1); in TEST()
53 TfLiteIntArray* b = TfLiteIntArrayCreate(2); in TEST()
56 TfLiteIntArray* c = TfLiteIntArrayCreate(2); in TEST()
59 TfLiteIntArray* d = TfLiteIntArrayCreate(2); in TEST()
114 params->zero_point = TfLiteIntArrayCreate(3); in TEST()
128 t.sparsity->traversal_order = TfLiteIntArrayCreate(2); in TEST()
139 t.sparsity->dim_metadata[1].array_segments = TfLiteIntArrayCreate(2); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_test.cc48 tflite_tensor.dims = TfLiteIntArrayCreate(1); in TEST()
63 tflite_tensor.dims = TfLiteIntArrayCreate(2); in TEST()
79 tflite_tensor.dims = TfLiteIntArrayCreate(3); in TEST()
96 tflite_tensor.dims = TfLiteIntArrayCreate(4); in TEST()
114 tflite_tensor.dims = TfLiteIntArrayCreate(0); in TEST()
127 tflite_tensor.dims = TfLiteIntArrayCreate(5); in TEST()
138 exec_plan_ = TfLiteIntArrayCreate(num_nodes); in DelegatedInterpreter()
178 TfLiteIntArray* new_array = TfLiteIntArrayCreate(num_nodes); in exec_plan()
336 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
338 params->input_tensors = TfLiteIntArrayCreate(2); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
H A Dtopk_accuracy_eval_stage_test.cc75 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
91 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
109 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
125 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
141 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
156 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
173 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
194 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
251 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in VerifyCorrectBehaviorForType()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/
H A Dop_signature_test.cc37 exec_plan_ = TfLiteIntArrayCreate(3);
45 nodes_[0].inputs = TfLiteIntArrayCreate(1);
47 nodes_[0].outputs = TfLiteIntArrayCreate(1);
52 nodes_[1].inputs = TfLiteIntArrayCreate(num_inputs);
56 nodes_[1].outputs = TfLiteIntArrayCreate(1);
62 nodes_[2].inputs = TfLiteIntArrayCreate(1);
64 nodes_[2].outputs = TfLiteIntArrayCreate(1);
74 tensors_[i].dims = TfLiteIntArrayCreate(4);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils_test.cc32 tensors[0].dims = TfLiteIntArrayCreate(2); in TEST()
81 exec_plan_ = TfLiteIntArrayCreate(10);
86 params1.nodes_to_replace = TfLiteIntArrayCreate(1);
91 params2.nodes_to_replace = TfLiteIntArrayCreate(4);
99 params3.nodes_to_replace = TfLiteIntArrayCreate(3);
106 params4.nodes_to_replace = TfLiteIntArrayCreate(2);
H A Dutils.cc42 TfLiteIntArray* dims = TfLiteIntArrayCreate(original_dims->size); in CreateNewTensorWithDifferentType()
129 original_execution_plan_ = TfLiteIntArrayCreate(execution_plan->size); in PrepareSupportedNodes()
133 supported_nodes_ = TfLiteIntArrayCreate(num_total_nodes_); in PrepareSupportedNodes()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dutils_test.cc28 TfLiteIntArray* shape_4d = TfLiteIntArrayCreate(4); in TEST()
46 TfLiteIntArray* shape_2d = TfLiteIntArrayCreate(2); in TEST()
62 TfLiteIntArray* shape_5d = TfLiteIntArrayCreate(5); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dmodel_test.cc240 TfLiteIntArray* desired_inputs = TfLiteIntArrayCreate(2); in TEST()
243 TfLiteIntArray* desired_outputs = TfLiteIntArrayCreate(1); in TEST()
259 TfLiteIntArray* desired_inputs = TfLiteIntArrayCreate(1); in TEST()
260 TfLiteIntArray* desired_outputs = TfLiteIntArrayCreate(1); in TEST()
606 TfLiteIntArray* traversal_order = TfLiteIntArrayCreate(4); in TEST()
615 TfLiteIntArray* block_map = TfLiteIntArrayCreate(2); in TEST()
630 TfLiteIntArray* array_segments = TfLiteIntArrayCreate(3); in TEST()
638 TfLiteIntArray* array_indices = TfLiteIntArrayCreate(3); in TEST()
H A Dstring_util_test.cc70 auto new_shape = TfLiteIntArrayCreate(2); in TEST()
172 t0->dims = TfLiteIntArrayCreate(2); in TEST()
186 auto new_shape = TfLiteIntArrayCreate(2); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/gradient/
H A Dbcast_grad_args.cc102 TfLiteIntArray* output1_shape = TfLiteIntArrayCreate(1); in Invoke()
107 TfLiteIntArray* output2_shape = TfLiteIntArrayCreate(1); in Invoke()
190 TfLiteIntArray* output1_shape = TfLiteIntArrayCreate(1); in Invoke()
204 TfLiteIntArray* output2_shape = TfLiteIntArrayCreate(1); in Invoke()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dwhitespace_tokenizer.cc95 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(NumDimensions(input) + 1); in WritePaddedOutput()
134 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(1); in WriteRaggedOutput()
171 TfLiteIntArray* row_splits_shape = TfLiteIntArrayCreate(1); in Prepare()

123456