Searched refs:n_cell_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | optional_tensor_test.cc | 39 n_cell_(n_cell), in LSTMOpModel()
102 AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, true); in LSTMOpModel()
193 int num_cells() { return n_cell_; } in num_cells()
226 int n_cell_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
| D | unidirectional_sequence_lstm_test.cc | 43 n_cell_(n_cell), in UnidirectionalLSTMOpModel()
108 AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, in UnidirectionalLSTMOpModel()
199 int num_cells() { return n_cell_; } in num_cells()
234 int n_cell_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
|
| D | lstm_test.cc | 45 n_cell_(n_cell), in LSTMOpModel()
108 AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, true); in LSTMOpModel()
225 int num_cells() { return n_cell_; } in num_cells()
265 int n_cell_; member in tflite::__anonb3707f480111::LSTMOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2266 n_cell_(n_cell), in LSTMOpModel()
2330 AddInput(TensorData{TensorType_FLOAT32, {n_batch_, n_cell_}}, true); in LSTMOpModel()
2418 int num_cells() { return n_cell_; } in num_cells()
2453 int n_cell_; member in tflite::__anon0c68fcb70111::LSTMOpModel
|