Searched defs:n_batch (Results 1 – 25 of 54) sorted by relevance
123
76 LSTMOpModel(uint32_t n_batch, uint32_t n_input, in LSTMOpModel()274 const int n_batch = 1; in TEST() local385 const int n_batch = 1; in TEST() local490 const int n_batch = 2; in TEST() local
87 LayerNormLSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LayerNormLSTMOpModel()287 const int n_batch = 2; in TEST() local
20 n_batch = 2 variable99 n_batch = 1 variable
22 n_batch = 2 variable194 n_batch = 2 variable
21 n_batch = 1 variable
21 n_batch = 2 variable