Home
last modified time | relevance | path

Searched refs:golden_dims (Results 1 – 3 of 3) sorted by relevance

/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/
Dgather_test.cc92 const int golden_dims[] = {2, 2}; in TF_LITE_MICRO_TEST() local
96 output_data, golden_dims, golden_data); in TF_LITE_MICRO_TEST()
111 const int golden_dims[] = {2}; in TF_LITE_MICRO_TEST() local
115 output_data, golden_dims, golden_data); in TF_LITE_MICRO_TEST()
132 const int golden_dims[] = {0}; in TF_LITE_MICRO_TEST() local
136 output_data, golden_dims, golden_data); in TF_LITE_MICRO_TEST()
151 const int golden_dims[] = {1}; in TF_LITE_MICRO_TEST() local
155 output_data, golden_dims, golden_data); in TF_LITE_MICRO_TEST()
170 const int golden_dims[] = {1, 2}; in TF_LITE_MICRO_TEST() local
174 output_data, golden_dims, golden_data); in TF_LITE_MICRO_TEST()
[all …]
Dexpand_dims_test.cc84 int golden_dims[] = {1, 2, 2}; in TF_LITE_MICRO_TEST() local
87 axis_data, golden_dims, output_dims, in TF_LITE_MICRO_TEST()
98 int golden_dims[] = {2, 1, 2}; in TF_LITE_MICRO_TEST() local
101 axis_data, golden_dims, output_dims, in TF_LITE_MICRO_TEST()
112 int golden_dims[] = {2, 2, 1}; in TF_LITE_MICRO_TEST() local
115 axis_data, golden_dims, output_dims, in TF_LITE_MICRO_TEST()
126 int golden_dims[] = {1, 3, 1, 2}; in TF_LITE_MICRO_TEST() local
129 axis_data, golden_dims, output_dims, in TF_LITE_MICRO_TEST()
140 int golden_dims[] = {3, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
143 axis_data, golden_dims, output_dims, in TF_LITE_MICRO_TEST()
[all …]
Dreshape_test.cc170 int golden_dims[] = {}; in TF_LITE_MICRO_TEST() local
173 golden_output, golden_output_len, golden_dims, golden_dims_len, true); in TF_LITE_MICRO_TEST()
186 int golden_dims[] = {1, 1, 1, 1, 1, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
189 golden_output, golden_output_len, golden_dims, golden_dims_len, false); in TF_LITE_MICRO_TEST()
202 int golden_dims[] = {}; in TF_LITE_MICRO_TEST() local
205 golden_output, golden_output_len, golden_dims, golden_dims_len, true); in TF_LITE_MICRO_TEST()
249 int golden_dims[] = {2, 2, 2}; in TF_LITE_MICRO_TEST() local
253 golden_dims, golden_dims_len, false); in TF_LITE_MICRO_TEST()
256 output_data_int8, golden_output_int8, golden_output_len, golden_dims, in TF_LITE_MICRO_TEST()
260 output_data_uint8, golden_output_uint8, golden_output_len, golden_dims, in TF_LITE_MICRO_TEST()
[all …]