Home
last modified time | relevance | path

Searched refs:n_batch_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc37 : n_batch_(n_batch), in LSTMOpModel()
100 AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, true); in LSTMOpModel()
102 AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, true); in LSTMOpModel()
194 int num_batches() { return n_batch_; } in num_batches()
224 int n_batch_; member in tflite::__anon67d22bc80111::LSTMOpModel
Dbidirectional_sequence_lstm_test.cc44 : n_batch_(n_batch), in BidirectionalLSTMOpModel()
168 AddInput(TensorData{TensorType_FLOAT32, {n_fw_output_ * n_batch_}}, in BidirectionalLSTMOpModel()
171 AddInput(TensorData{TensorType_FLOAT32, {n_fw_cell_ * n_batch_}}, in BidirectionalLSTMOpModel()
176 AddInput(TensorData{TensorType_FLOAT32, {n_bw_output_ * n_batch_}}, in BidirectionalLSTMOpModel()
179 AddInput(TensorData{TensorType_FLOAT32, {n_bw_cell_ * n_batch_}}, in BidirectionalLSTMOpModel()
321 int num_batches() { return n_batch_; } in num_batches()
388 int n_batch_; member in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
Dunidirectional_sequence_lstm_test.cc41 : n_batch_(n_batch), in UnidirectionalLSTMOpModel()
105 AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, in UnidirectionalLSTMOpModel()
108 AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, in UnidirectionalLSTMOpModel()
200 int num_batches() { return n_batch_; } in num_batches()
232 int n_batch_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
Dlstm_test.cc43 : n_batch_(n_batch), in LSTMOpModel()
106 AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, true); in LSTMOpModel()
108 AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, true); in LSTMOpModel()
226 int num_batches() { return n_batch_; } in num_batches()
263 int n_batch_; member in tflite::__anonb3707f480111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2264 : n_batch_(n_batch), in LSTMOpModel()
2328 AddInput(TensorData{TensorType_FLOAT32, {n_batch_, n_output_}}, true); in LSTMOpModel()
2330 AddInput(TensorData{TensorType_FLOAT32, {n_batch_, n_cell_}}, true); in LSTMOpModel()
2419 int num_batches() { return n_batch_; } in num_batches()
2451 int n_batch_; member in tflite::__anon0c68fcb70111::LSTMOpModel