Searched defs:n_batch (Results 1 – 23 of 23) sorted by relevance
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
246 const uint32_t n_batch = SizeOfDimension(input, 0); in Prepare() local303 const uint32_t n_batch = input_->shape().dimensions[0]; in Eval() local
21 n_batch = 1 variable
21 n_batch = 2 variable