Searched refs:rnn_input (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn_test.cc | 34 static float rnn_input[] = { variable
264 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST()
268 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
291 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST()
295 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
319 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST()
323 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
|
| D | unidirectional_sequence_rnn_test.cc | 32 static float rnn_input[] = { variable
284 float* batch_start = rnn_input; in TEST()
309 float* batch_start = rnn_input; in TEST()
335 float* batch_start = rnn_input; in TEST()
361 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
390 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
420 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
|
| D | bidirectional_sequence_rnn_test.cc | 32 static float rnn_input[] = { variable
799 float* batch_start = rnn_input; in TEST()
841 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
875 float* batch_start = rnn_input; in TEST()
917 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
961 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
1082 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
1125 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
1171 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | unidirectional_sequence_rnn_test.py | 80 rnn_input = tf.transpose(x, perm=[1, 0, 2])
82 rnn_layer, rnn_input, dtype="float32")
85 rnn_input = tf.unstack(x, self.time_steps, 1)
86 outputs, _ = tf.nn.static_rnn(rnn_layer, rnn_input, dtype="float32")
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1761 static float rnn_input[] = { variable
1966 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST()
1970 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
|