Searched defs:batch_start (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn_test.cc | 799 float* batch_start = rnn_input; in TEST() local
841 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
875 float* batch_start = rnn_input; in TEST() local
917 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
961 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1029 float* batch_start = endtoend_input + k * input_sequence_size; in TEST() local
1082 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1125 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1171 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
|
| D | unidirectional_sequence_rnn_test.cc | 284 float* batch_start = rnn_input; in TEST() local
309 float* batch_start = rnn_input; in TEST() local
335 float* batch_start = rnn_input; in TEST() local
361 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
390 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
420 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
|
| D | basic_rnn_test.cc | 268 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
295 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
323 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
|
| D | svdf_test.cc | 245 float* batch_start = in VerifyGoldens() local
|
| D | unidirectional_sequence_lstm_test.cc | 355 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
364 const float* batch_start = input[b].data(); in VerifyGoldens() local
|
| D | lstm_test.cc | 379 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
1777 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
|
| D | fully_connected_test.cc | 862 float* batch_start = fully_connected_input + i * m.input_size(); in TEST_P() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils.py | 76 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/libtextclassifier/annotator/ |
| D | annotator.cc | 2168 for (int batch_start = span_of_interest.first; in ModelClickContextScoreChunks() local
2279 for (int batch_start = 0; batch_start < candidate_spans.size(); in ModelBoundsSensitiveScoreChunks() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1970 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
2174 float* batch_start = in VerifyGoldens() local
2497 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.cc | 1009 &output](int64 batch_start, int64 batch_limit) { in operator ()()
|