| /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 | 192 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
223 if (!TfLiteIntArrayEqualsArray(zero_points->dims, 1, zero_points_dims)) { in Prepare()
237 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1045 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1059 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1073 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1093 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1111 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
1126 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1140 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1159 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 2, row_sums_dims)) { in Prepare()
1195 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 | 291 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in PrepareImpl()
305 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in PrepareImpl()
320 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in PrepareImpl()
333 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in PrepareImpl()
|
| D | conv.cc | 554 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
570 if (!TfLiteIntArrayEqualsArray(accum_scratch->dims, 2, in Prepare()
597 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, in Prepare()
613 if (!TfLiteIntArrayEqualsArray(row_sums->dims, 1, row_sums_dims)) { in Prepare()
|
| D | depthwise_conv.cc | 255 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
268 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 | 109 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size,
|