Searched refs:n_output_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | optional_tensor_test.cc | 40 n_output_(n_output) { in LSTMOpModel()
100 AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, true); in LSTMOpModel()
192 int num_outputs() { return n_output_; } in num_outputs()
227 int n_output_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
| D | unidirectional_sequence_lstm_test.cc | 44 n_output_(n_output), in UnidirectionalLSTMOpModel()
105 AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, in UnidirectionalLSTMOpModel()
198 int num_outputs() { return n_output_; } in num_outputs()
235 int n_output_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
|
| D | lstm_test.cc | 46 n_output_(n_output) { in LSTMOpModel()
106 AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, true); in LSTMOpModel()
224 int num_outputs() { return n_output_; } in num_outputs()
266 int n_output_; member in tflite::__anonb3707f480111::LSTMOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2267 n_output_(n_output), in LSTMOpModel()
2328 AddInput(TensorData{TensorType_FLOAT32, {n_batch_, n_output_}}, true); in LSTMOpModel()
2417 int num_outputs() { return n_output_; } in num_outputs()
2454 int n_output_; member in tflite::__anon0c68fcb70111::LSTMOpModel
|