| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_internal_test.cc | 27 TfLiteIntArray* a = TfLiteIntArrayCreate(0); in TEST()
28 TfLiteIntArray* b = TfLiteIntArrayCreate(3); in TEST()
34 TfLiteIntArray* a = TfLiteIntArrayCreate(2); in TEST()
47 TfLiteIntArray* a = TfLiteIntArrayCreate(1); in TEST()
49 TfLiteIntArray* b = TfLiteIntArrayCreate(2); in TEST()
52 TfLiteIntArray* c = TfLiteIntArrayCreate(2); in TEST()
55 TfLiteIntArray* d = TfLiteIntArrayCreate(2); in TEST()
100 params->zero_point = TfLiteIntArrayCreate(3); in TEST()
|
| D | c_api_internal.c | 45 TfLiteIntArray* TfLiteIntArrayCreate(int size) { in TfLiteIntArrayCreate() function
64 TfLiteIntArray* ret = TfLiteIntArrayCreate(src->size); in TfLiteIntArrayCopy()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 45 tensor->dims = TfLiteIntArrayCreate(dims.size()); in SetShape()
153 input.dims = TfLiteIntArrayCreate(1); in TEST_F()
162 input_params->zero_point = TfLiteIntArrayCreate(1); in TEST_F()
170 filter.dims = TfLiteIntArrayCreate(4); in TEST_F()
184 filter_params->zero_point = TfLiteIntArrayCreate(3); in TEST_F()
195 bias.dims = TfLiteIntArrayCreate(4); in TEST_F()
205 bias_params->zero_point = TfLiteIntArrayCreate(3); in TEST_F()
223 output_params->zero_point = TfLiteIntArrayCreate(1); in TEST_F()
259 input.dims = TfLiteIntArrayCreate(1); in TEST_F()
268 input_params->zero_point = TfLiteIntArrayCreate(1); in TEST_F()
[all …]
|
| D | reduce.cc | 78 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis()
86 TfLiteIntArray* size = TfLiteIntArrayCreate(1); in ResizeTempSum()
98 TfLiteIntArrayCreate(0)); in ResizeOutputTensor()
102 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims); in ResizeOutputTensor()
140 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis); in ResizeOutputTensor()
165 node->temporaries = TfLiteIntArrayCreate(3); in InitializeTemporaries()
170 TfLiteIntArray* index_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
|
| D | svdf.cc | 172 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
186 node->temporaries = TfLiteIntArrayCreate(4); in Prepare()
188 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
192 TfLiteIntArray* scratch_size_array = TfLiteIntArrayCreate(2); in Prepare()
222 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
|
| D | lstm.cc | 377 TfLiteIntArray* output_size = TfLiteIntArrayCreate(2); in Prepare()
391 node->temporaries = TfLiteIntArrayCreate(7); in Prepare()
393 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
405 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
466 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
478 TfLiteIntArray* prod_scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
495 TfLiteIntArray* recovered_cell_weights_size = TfLiteIntArrayCreate(1); in Prepare()
714 TfLiteIntArray* concat_temp_size = TfLiteIntArrayCreate(2); in Prepare()
719 TfLiteIntArray* activation_temp_size = TfLiteIntArrayCreate(2); in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 315 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
317 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
329 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
396 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
409 TfLiteIntArray* prod_scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
427 TfLiteIntArray* recovered_cell_weights_size = TfLiteIntArrayCreate(1); in Prepare()
|
| D | transpose_conv.cc | 83 TfLiteIntArray* shape = TfLiteIntArrayCreate(NumElements(shape_tensor)); in ResizeOutputTensor()
100 node->temporaries = TfLiteIntArrayCreate(1); in AllocateIm2colTensorIfRequired()
117 TfLiteIntArray* im2col_shape_array = TfLiteIntArrayCreate(4); in ResizeIm2ColTensor()
|
| D | bidirectional_sequence_lstm.cc | 503 TfLiteIntArray* fw_output_size = TfLiteIntArrayCreate(3); in Prepare()
517 node->temporaries = TfLiteIntArrayCreate( in Prepare()
520 node->temporaries = TfLiteIntArrayCreate(2); // the two scratch buffers. in Prepare()
536 TfLiteIntArray* fw_scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
563 TfLiteIntArray* bw_output_size = TfLiteIntArrayCreate(3); in Prepare()
593 TfLiteIntArray* bw_scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
689 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
702 TfLiteIntArray* prod_scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
720 TfLiteIntArray* recovered_cell_weights_size = TfLiteIntArrayCreate(1); in Prepare()
|
| D | basic_rnn.cc | 92 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
107 node->temporaries = TfLiteIntArrayCreate(3); in Prepare()
136 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
|
| D | hashtable_lookup.cc | 80 TfLiteIntArray* hitSize = TfLiteIntArrayCreate(1); in Prepare()
88 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
|
| D | reshape.cc | 66 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(shape->dims->data[0]); in GetOutputShapeFromTensor()
87 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam()
|
| D | unidirectional_sequence_rnn.cc | 92 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(3); in Prepare()
108 node->temporaries = TfLiteIntArrayCreate(3); in Prepare()
137 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
|
| D | rank.cc | 37 TfLiteIntArray* output_size = TfLiteIntArrayCreate(0); in Prepare()
|
| D | bidirectional_sequence_rnn.cc | 180 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
183 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors - 1); in Prepare()
237 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
262 TfLiteIntArray* fw_output_size_array = TfLiteIntArrayCreate(3); in Prepare()
271 TfLiteIntArray* bw_output_size_array = TfLiteIntArrayCreate(3); in Prepare()
|
| D | conv.cc | 210 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in AllocateTemporaryTensorsIfRequired()
327 TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); in Prepare()
339 TfLiteIntArray* im2col_size = TfLiteIntArrayCreate(4); in Prepare()
360 TfLiteIntArray* hwcn_weights_size = TfLiteIntArrayCreate(2); in Prepare()
409 TfLiteIntArray* scaling_factors_size = TfLiteIntArrayCreate(1); in Prepare()
|
| D | topk_v2.cc | 49 TfLiteIntArray* output_indexes_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
50 TfLiteIntArray* output_values_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
|
| D | shape.cc | 59 TfLiteIntArray* output_size = TfLiteIntArrayCreate(1); in Prepare()
|
| /external/tensorflow/tensorflow/lite/ |
| D | string_util_test.cc | 57 auto new_shape = TfLiteIntArrayCreate(2); in TEST()
133 t0->dims = TfLiteIntArrayCreate(2); in TEST()
147 auto new_shape = TfLiteIntArrayCreate(2); in TEST()
|
| D | model_test.cc | 167 TfLiteIntArray* desired_inputs = TfLiteIntArrayCreate(2); in TEST()
170 TfLiteIntArray* desired_outputs = TfLiteIntArrayCreate(1); in TEST()
186 TfLiteIntArray* desired_inputs = TfLiteIntArrayCreate(1); in TEST()
187 TfLiteIntArray* desired_outputs = TfLiteIntArrayCreate(1); in TEST()
|
| D | util.cc | 31 TfLiteIntArray* output = TfLiteIntArrayCreate(rank); in ConvertArrayToTfLiteIntArray()
|
| /external/tensorflow/tensorflow/lite/models/smartreply/ops/ |
| D | extract_feature.cc | 62 TfLiteIntArray* outputSize1 = TfLiteIntArrayCreate(1); in Prepare()
63 TfLiteIntArray* outputSize2 = TfLiteIntArrayCreate(1); in Prepare()
|
| D | predict.cc | 105 TfLiteIntArray* label_size = TfLiteIntArrayCreate(1); in Prepare()
107 TfLiteIntArray* weight_size = TfLiteIntArrayCreate(1); in Prepare()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 91 TfLiteIntArray* log_probability_output_shape_array = TfLiteIntArrayCreate(2); in Prepare()
102 TfLiteIntArray* output_shape_array = TfLiteIntArrayCreate(dimensions); in Resize()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 41 TfLiteIntArray* shape = TfLiteIntArrayCreate(num_dims); in CopyShapeAndType()
|