Home
last modified time | relevance | path

Searched defs:n_batch_ (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm_test_util.h39 : n_batch_(n_batch), in n_batch_() function
265 int n_batch_; variable
H A Dunidirectional_sequence_gru_test.cc94 int n_batch_; member in tflite::ops::custom::__anonf3f34e7d0111::GRUOpModel
H A Dlstm_test.cc275 int n_batch_; member in tflite::__anon299f347e0111::LSTMOpModel
2337 int n_batch_; member in tflite::__anon299f347e0111::HybridSparseLSTMOpModel
2376 int n_batch_; member in tflite::__anon299f347e0111::BaseSparseLstmTest
H A Doptional_tensor_test.cc222 int n_batch_; member in tflite::__anon499b733e0111::LSTMOpModel
H A Dlstm_eval_test.cc146 const int n_batch_ = 2; member in tflite::__anone9b10f650111::BaseLstmParam
H A Dbidirectional_sequence_lstm_test.cc409 int n_batch_; member in tflite::__anon4ad9bf6e0111::BidirectionalLSTMOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A Dlstm_full_test.cc191 int n_batch_; member in tflite::__anone147aac60111::LSTMOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc3402 int n_batch_; member in tflite::__anon1a74836d0111::LSTMOpModel