Home
last modified time | relevance | path

Searched defs:batch_end (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn_test.cc800 float* batch_end = batch_start + input_sequence_size; in TEST() local
842 float* batch_end = batch_start + rnn.input_size(); in TEST() local
876 float* batch_end = batch_start + input_sequence_size; in TEST() local
918 float* batch_end = batch_start + rnn.input_size(); in TEST() local
962 float* batch_end = batch_start + rnn.input_size(); in TEST() local
1030 float* batch_end = batch_start + input_sequence_size; in TEST() local
1083 float* batch_end = batch_start + rnn.input_size(); in TEST() local
1126 float* batch_end = batch_start + rnn.input_size(); in TEST() local
1172 float* batch_end = batch_start + rnn.input_size(); in TEST() local
Dunidirectional_sequence_rnn_test.cc285 float* batch_end = batch_start + input_sequence_size; in TEST() local
310 float* batch_end = batch_start + input_sequence_size; in TEST() local
336 float* batch_end = batch_start + input_sequence_size; in TEST() local
362 float* batch_end = batch_start + rnn.input_size(); in TEST() local
391 float* batch_end = batch_start + rnn.input_size(); in TEST() local
421 float* batch_end = batch_start + rnn.input_size(); in TEST() local
Dbasic_rnn_test.cc269 float* batch_end = batch_start + rnn.input_size(); in TEST() local
296 float* batch_end = batch_start + rnn.input_size(); in TEST() local
324 float* batch_end = batch_start + rnn.input_size(); in TEST() local
Dsvdf_test.cc247 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in VerifyGoldens() local
Dunidirectional_sequence_lstm_test.cc356 const float* batch_end = batch_start + num_inputs; in VerifyGoldens() local
365 const float* batch_end = batch_start + input_sequence_size * num_inputs; in VerifyGoldens() local
Dlstm_test.cc380 const float* batch_end = batch_start + num_inputs; in VerifyGoldens() local
1778 const float* batch_end = batch_start + num_inputs; in VerifyGoldens() local
Dfully_connected_test.cc863 float* batch_end = batch_start + m.input_size(); in TEST_P() local
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils.py76 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
100 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
130 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
/external/tensorflow/tensorflow/core/kernels/
Dmatrix_band_part_op.cc154 const int64 batch_end = (end + m - 1) / m; in operator ()() local
/external/libtextclassifier/annotator/
Dannotator.cc2170 const int batch_end = in ModelClickContextScoreChunks() local
2281 const int batch_end = std::min(batch_start + max_batch_size, in ModelBoundsSensitiveScoreChunks() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1971 float* batch_end = batch_start + rnn.input_size(); in TEST() local
2176 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in VerifyGoldens() local
2498 const float* batch_end = batch_start + num_inputs; in VerifyGoldens() local