Searched defs:n_input (Results 1 – 23 of 23) sorted by relevance
83 uint32_t n_input, uint32_t n_output, uint32_t n_cell) { in CheckInputTensorDimensions()247 const uint32_t n_input = SizeOfDimension(input, 1); in Prepare() local304 const uint32_t n_input = input_->shape().dimensions[1]; in Eval() local
76 LSTMOpModel(uint32_t n_batch, uint32_t n_input, in LSTMOpModel()275 const int n_input = 2; in TEST() local386 const int n_input = 2; in TEST() local491 const int n_input = 5; in TEST() local
22 n_input = 2 variable
22 n_input = 5 variable