| /external/tensorflow/tensorflow/lite/kernels/ |
| basic_rnn_test.cc | 269 float* batch_end = batch_start + rnn.input_size(); local
270 rnn.SetInput(0, batch_start, batch_end);
271 rnn.SetInput(rnn.input_size(), batch_start, batch_end);
296 float* batch_end = batch_start + rnn.input_size(); local
297 rnn.SetInput(0, batch_start, batch_end);
298 rnn.SetInput(rnn.input_size(), batch_start, batch_end);
324 float* batch_end = batch_start + rnn.input_size(); local
325 rnn.SetInput(0, batch_start, batch_end);
326 rnn.SetInput(rnn.input_size(), batch_start, batch_end);
|
| unidirectional_sequence_rnn_test.cc | 285 float* batch_end = batch_start + input_sequence_size; local
286 rnn.SetInput(0, batch_start, batch_end);
287 rnn.SetInput(input_sequence_size, batch_start, batch_end);
310 float* batch_end = batch_start + input_sequence_size; local
311 rnn.SetInput(0, batch_start, batch_end);
312 rnn.SetInput(input_sequence_size, batch_start, batch_end);
336 float* batch_end = batch_start + input_sequence_size; local
337 rnn.SetInput(0, batch_start, batch_end);
338 rnn.SetInput(input_sequence_size, batch_start, batch_end);
362 float* batch_end = batch_start + rnn.input_size() local
391 float* batch_end = batch_start + rnn.input_size(); local
421 float* batch_end = batch_start + rnn.input_size(); local
[all...] |
| svdf_test.cc | 247 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; local
248 svdf->SetInput(0, batch_start, batch_end);
|
| bidirectional_sequence_rnn_test.cc | 800 float* batch_end = batch_start + input_sequence_size; local
842 float* batch_end = batch_start + rnn.input_size(); local
876 float* batch_end = batch_start + input_sequence_size; local
918 float* batch_end = batch_start + rnn.input_size(); local
962 float* batch_end = batch_start + rnn.input_size(); local
1030 float* batch_end = batch_start + input_sequence_size; local
1083 float* batch_end = batch_start + rnn.input_size(); local
1126 float* batch_end = batch_start + rnn.input_size(); local
1172 float* batch_end = batch_start + rnn.input_size(); local
[all...] |
| fully_connected_test.cc | 863 float* batch_end = batch_start + m.input_size(); local
864 m.SetInput(0, batch_start, batch_end);
865 m.SetInput(m.input_size(), batch_start, batch_end);
|
| unidirectional_sequence_lstm_test.cc | 356 const float* batch_end = batch_start + num_inputs; local
359 batch_end);
365 const float* batch_end = batch_start + input_sequence_size * num_inputs; local
368 batch_end);
[all...] |
| lstm_test.cc | 380 const float* batch_end = batch_start + num_inputs; local
382 lstm->SetInput(b * lstm->num_inputs(), batch_start, batch_end);
1778 const float* batch_end = batch_start + num_inputs; local
[all...] |
| /external/libtextclassifier/annotator/ |
| annotator.cc | 2170 const int batch_end = local
2281 const int batch_end = std::min(batch_start + max_batch_size, local
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | 1971 float* batch_end = batch_start + rnn.input_size(); local
2176 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; local
2498 const float* batch_end = batch_start + num_inputs; local
[all...] |
