| /external/tensorflow/tensorflow/lite/kernels/ |
| 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()
189 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
|
| D | bidirectional_sequence_lstm.cc | 755 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
769 if (!TfLiteIntArrayEqualsArray(aux_input_sf->dims, 1, scaling_dims)) { in Prepare()
783 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
797 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
817 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
841 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
858 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
872 if (!TfLiteIntArrayEqualsArray(aux_input_zp->dims, 1, scaling_dims)) { in Prepare()
886 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
912 if (!TfLiteIntArrayEqualsArray(fw_row_sums->dims, 2, fw_row_sums_dims)) { in Prepare()
[all …]
|
| 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()
|
| D | svdf.cc | 191 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
222 if (!TfLiteIntArrayEqualsArray(zero_points->dims, 1, zero_points_dims)) { in Prepare()
236 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1036 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1050 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1064 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1084 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1102 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
1117 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1131 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1150 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
1186 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
|
| 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()
304 if (!TfLiteIntArrayEqualsArray(fw_row_sums->dims, 2, fw_row_sums_dims)) { in Prepare()
320 if (!TfLiteIntArrayEqualsArray(bw_row_sums->dims, 2, bw_row_sums_dims)) { in Prepare()
|
| D | batch_matmul.cc | 238 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in InitializeTemporaries()
252 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in InitializeTemporaries()
267 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in InitializeTemporaries()
280 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in InitializeTemporaries()
|
| D | lstm.cc | 1494 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1508 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1522 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1542 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1560 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
1575 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1589 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1614 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
1770 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
1808 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
|
| D | fully_connected.cc | 290 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in PrepareImpl()
304 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in PrepareImpl()
319 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in PrepareImpl()
332 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in PrepareImpl()
|
| D | conv.cc | 529 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
545 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
571 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, in Prepare()
587 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in Prepare()
|
| D | depthwise_conv.cc | 253 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
266 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in Prepare()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.c | 32 return TfLiteIntArrayEqualsArray(a, b->size, b->data); in TfLiteIntArrayEqual()
35 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size, in TfLiteIntArrayEqualsArray() function
|
| D | common.h | 108 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size,
|