| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 754 int scaling_dims[1] = {n_batch}; in Prepare() local
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()
798 scaling_dims)) { 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()
|
| D | basic_rnn.cc | 147 int scaling_dims[1] = {batch_size}; in Prepare() local
148 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
|
| D | svdf.cc | 190 int scaling_dims[1] = {batch_size}; in Prepare() local
191 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
|
| D | depthwise_conv.cc | 252 int scaling_dims[1] = {batch_size}; in Prepare() local
253 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
266 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1035 int scaling_dims[1] = {n_batch}; in Prepare() local
1036 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1050 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1065 scaling_dims)) { 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()
|
| D | unidirectional_sequence_rnn.cc | 154 int scaling_dims[1] = {batch_size}; in Prepare() local
155 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
|
| D | batch_matmul.cc | 237 int scaling_dims[1] = {num_batches * batch_size}; in InitializeTemporaries() local
238 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in InitializeTemporaries()
267 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in InitializeTemporaries()
|
| D | fully_connected.cc | 289 int scaling_dims[1] = {batch_size}; in PrepareImpl() local
290 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in PrepareImpl()
319 if (!TfLiteIntArrayEqualsArray(input_offsets->dims, 1, scaling_dims)) { in PrepareImpl()
|
| D | lstm.cc | 1493 int scaling_dims[1] = {n_batch}; in Prepare() local
1494 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1508 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1523 scaling_dims)) { 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()
|
| D | bidirectional_sequence_rnn.cc | 255 int scaling_dims[1] = {batch_size}; in Prepare() local
256 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
|
| D | conv.cc | 528 int scaling_dims[1] = {height}; in Prepare() local
529 if (!TfLiteIntArrayEqualsArray(scaling_factors->dims, 1, scaling_dims)) { in Prepare()
|