Home
last modified time | relevance | path

Searched defs:LSTMOpModel (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Doptional_tensor_test.cc29 class LSTMOpModel : public SingleOpModel { class
31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anon499b733e0111::LSTMOpModel
H A Dlstm_test.cc37 class LSTMOpModel : public SingleOpModel { class
39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anon299f347e0111::LSTMOpModel
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp76 class LSTMOpModel { class
78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel() function in android::nn::wrapper::LSTMOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A Dlstm_full_test.cc33 class LSTMOpModel : public SingleOpModel { class
35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anone147aac60111::LSTMOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc3167 class LSTMOpModel : public SingleOpModelWithNNAPI { class
3169 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() function in tflite::__anon1a74836d0111::LSTMOpModel