| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
271 HybridLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in HybridLSTMOpModel()
453 const int n_cell = 4; in TEST_F() local
510 const int n_cell = 4; in TEST_F() local
569 const int n_cell = 4; in TEST_F() local
674 const int n_cell = 4; in TEST_F() local
731 const int n_cell = 4; in TEST_F() local
790 const int n_cell = 4; in TEST_F() local
1446 const int n_cell = 20; in TEST_F() local
1507 const int n_cell = 20; in TEST_F() local
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 34 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in UnidirectionalLSTMOpModel()
243 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
454 const int n_cell = 4; in TEST_F() local
516 const int n_cell = 4; in TEST_F() local
583 const int n_cell = 4; in TEST_F() local
646 const int n_cell = 4; in TEST_F() local
755 const int n_cell = 4; in TEST_F() local
817 const int n_cell = 4; in TEST_F() local
879 const int n_cell = 4; in TEST_F() local
1539 const int n_cell = 20; in TEST_F() local
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 37 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
409 const int n_cell = 4; in TEST_P() local
574 const int n_cell = 4; in TEST_P() local
739 const int n_cell = 4; in TEST() local
902 const int n_cell = 4; in TEST() local
1055 const int n_cell = 4; in TEST() local
1208 const int n_cell = 20; in TEST() local
1913 const int n_cell = 20; in TEST() local
2615 const int n_cell = 4; in TEST_P() local
|
| D | optional_tensor_test.cc | 33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
234 const int n_cell = 4; in TEST() local
|
| D | unidirectional_sequence_lstm.cc | 102 int n_output, int n_cell) { in CheckInputTensorDimensions()
272 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare() local
|
| D | lstm_eval.cc | 116 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
430 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
912 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat() local
1115 const int n_cell = input_to_output_weights->dims->data[0]; in EvalHybrid() local
|
| D | lstm.cc | 118 int n_output, int n_cell, in CheckInputTensorDimensions()
346 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare() local
|
| D | bidirectional_sequence_lstm.cc | 159 int n_cell, int input_to_input_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
351 int n_output, int n_cell) { in CheckInputTensorDimensions()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2259 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
2571 const int n_cell = 4; in TEST_F() local
2675 const int n_cell = 4; in TEST_F() local
3330 const int n_cell = 20; in TEST_F() local
|