| /external/tensorflow/tensorflow/lite/kernels/ |
| D | optional_tensor_test.cc | 35 : n_batch_(n_batch), in LSTMOpModel()
98 TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}); in LSTMOpModel()
100 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in LSTMOpModel()
192 int num_batches() { return n_batch_; } in num_batches()
222 int n_batch_; member in tflite::__anon4dd1434f0111::LSTMOpModel
|
| D | lstm_eval_test.cc | 146 const int n_batch_ = 2; member in tflite::__anon39d466b60111::BaseLstmParam
151 std::vector<int32_t> input_size_ = {n_batch_, n_input_};
310 std::vector<int32_t> activation_size_ = {n_batch_, n_output_};
313 std::vector<int32_t> cell_size_ = {n_batch_, n_cell_};
316 std::vector<int32_t> output_size_ = {n_batch_, n_output_};
585 std::vector<int32_t> scratch0_size_ = {n_batch_, n_cell_};
588 std::vector<int32_t> scratch1_size_ = {n_batch_, n_cell_};
591 std::vector<int32_t> scratch2_size_ = {n_batch_, n_cell_};
594 std::vector<int32_t> scratch3_size_ = {n_batch_, n_cell_};
597 std::vector<int32_t> scratch4_size_ = {n_batch_, n_cell_};
[all …]
|
| D | lstm_test.cc | 46 n_batch_(n_batch), in LSTMOpModel()
240 int num_batches() { return n_batch_; } in num_batches()
275 int n_batch_; member in tflite::__anonfa79752f0111::LSTMOpModel
2145 : n_batch_(n_batch), in HybridSparseLSTMOpModel()
2218 ::tflite::TensorType_FLOAT32, {n_output_ * n_batch_}}); in HybridSparseLSTMOpModel()
2220 ::tflite::TensorType_FLOAT32, {n_cell_ * n_batch_}}); in HybridSparseLSTMOpModel()
2301 int num_batches() { return n_batch_; } in num_batches()
2337 int n_batch_; member in tflite::__anonfa79752f0111::HybridSparseLSTMOpModel
2376 int n_batch_; member in tflite::__anonfa79752f0111::BaseSparseLstmTest
2426 n_batch_ = 2; in SetUp()
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 40 : n_batch_(n_batch), in BidirectionalLSTMOpModel()
164 TensorData{TensorType_FLOAT32, {n_fw_output_ * n_batch_}}); in BidirectionalLSTMOpModel()
166 TensorData{TensorType_FLOAT32, {n_fw_cell_ * n_batch_}}); in BidirectionalLSTMOpModel()
170 TensorData{TensorType_FLOAT32, {n_bw_output_ * n_batch_}}); in BidirectionalLSTMOpModel()
172 TensorData{TensorType_FLOAT32, {n_bw_cell_ * n_batch_}}); in BidirectionalLSTMOpModel()
342 int num_batches() { return n_batch_; } in num_batches()
409 int n_batch_; member in tflite::__anonc39730ff0111::BidirectionalLSTMOpModel
|
| D | unidirectional_sequence_lstm_test.cc | 41 : n_batch_(n_batch), in UnidirectionalLSTMOpModel()
105 TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}); in UnidirectionalLSTMOpModel()
107 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in UnidirectionalLSTMOpModel()
230 int num_batches() { return n_batch_; } in num_batches()
267 int n_batch_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru_test.cc | 37 : n_batch_(n_batch), n_input_(n_input), n_output_(n_output) { in GRUOpModel()
80 int num_batches() { return n_batch_; } in num_batches()
94 int n_batch_; member in tflite::ops::experimental::__anoncf6e42eb0111::GRUOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 63 n_batch_(n_batch), in LSTMOpModel()
183 int num_batches() { return n_batch_; } in num_batches()
191 int n_batch_; member in tflite::__anonb664ae370111::LSTMOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3097 : n_batch_(n_batch), in LSTMOpModel()
3288 int num_batches() { return n_batch_; } in num_batches()
3325 int n_batch_; member in tflite::__anonde5f765e0111::LSTMOpModel
|