| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_lstm.cc | 759 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
773 if (!TfLiteIntArrayEqualsArray(aux_input_sf->dims, 1, scaling_dims)) { in Prepare()
787 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
801 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
821 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
845 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
862 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
876 if (!TfLiteIntArrayEqualsArray(aux_input_zp->dims, 1, scaling_dims)) { in Prepare()
890 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
916 if (!TfLiteIntArrayEqualsArray(fw_row_sums->dims, 2, fw_row_sums_dims)) { in Prepare()
[all …]
|
| H A D | basic_rnn.cc | 148 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
161 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
176 if (!TfLiteIntArrayEqualsArray(zero_points->dims, 1, zero_points_dims)) { in Prepare()
190 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
|
| H A D | unidirectional_sequence_rnn.cc | 155 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
168 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
183 if (!TfLiteIntArrayEqualsArray(zero_points->dims, 1, zero_points_dims)) { in Prepare()
196 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
|
| H A D | unidirectional_sequence_lstm.cc | 1048 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1062 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1076 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1096 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1114 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
1129 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1143 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1163 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
1199 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
|
| H A D | svdf.cc | 192 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
224 if (!TfLiteIntArrayEqualsArray(zero_points->dims, 1, zero_points_dims)) { in Prepare()
239 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in Prepare()
|
| H A D | bidirectional_sequence_rnn.cc | 256 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
271 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
288 if (!TfLiteIntArrayEqualsArray(zero_points->dims, 1, zero_points_dims)) { in Prepare()
305 if (!TfLiteIntArrayEqualsArray(fw_row_sums->dims, 2, fw_row_sums_dims)) { in Prepare()
322 if (!TfLiteIntArrayEqualsArray(bw_row_sums->dims, 2, bw_row_sums_dims)) { in Prepare()
|
| H A D | lstm.cc | 1497 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1511 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1525 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1545 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1563 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
1578 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1592 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1618 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
1774 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
1812 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
|
| H A D | batch_matmul.cc | 241 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in InitializeTemporaries()
255 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in InitializeTemporaries()
270 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in InitializeTemporaries()
283 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in InitializeTemporaries()
|
| H A D | conv.cc | 577 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
593 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
620 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, in Prepare()
637 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in Prepare()
|
| H A D | fully_connected.cc | 367 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in PrepareImpl()
381 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in PrepareImpl()
396 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in PrepareImpl()
409 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in PrepareImpl()
|
| H A D | depthwise_conv.cc | 257 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
270 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | common.cc | 44 return TfLiteIntArrayEqualsArray(a, b->size, b->data); in TfLiteIntArrayEqual()
47 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size, in TfLiteIntArrayEqualsArray() function
|
| H A D | common.h | 113 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size,
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | common.h | 122 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size,
|