Home
last modified time | relevance | path

Searched refs:LSTMOpModel (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc31 class LSTMOpModel : public SingleOpModel { class
33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anon67d22bc80111::LSTMOpModel
237 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
Dlstm_test.cc35 class LSTMOpModel : public SingleOpModel { class
37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anonb3707f480111::LSTMOpModel
269 class HybridLSTMOpModel : public LSTMOpModel {
277 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole, in HybridLSTMOpModel()
370 LSTMOpModel* lstm, float tolerance = 1e-5) { in VerifyGoldens()
456 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
677 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
1449 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
1631 class LayerNormLSTMOpModel : public LSTMOpModel {
1639 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole, in LayerNormLSTMOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2257 class LSTMOpModel : public SingleOpModelWithNNAPI { class
2259 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anon0c68fcb70111::LSTMOpModel
2488 LSTMOpModel* lstm, float tolerance = 1e-5) { in VerifyGoldens()
2574 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
2678 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
3333 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()