Lines Matching refs:n_input
78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel() argument
82 : n_input_(n_input), in LSTMOpModel()
137 Input_.insert(Input_.end(), n_batch * n_input, 0.f); in LSTMOpModel()
274 const int n_input = 2; in TEST() local
279 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
285 {n_batch, n_input}, // input tensor in TEST()
287 {n_cell, n_input}, // input_to_input_weight tensor in TEST()
288 {n_cell, n_input}, // input_to_forget_weight tensor in TEST()
289 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
290 {n_cell, n_input}, // input_to_output_weight tensor in TEST()
378 const int n_input = 2; in TEST() local
383 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
389 {n_batch, n_input}, // input tensor in TEST()
392 {n_cell, n_input}, // input_to_forget_weight tensor in TEST()
393 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
394 {n_cell, n_input}, // input_to_output_weight tensor in TEST()
475 const int n_input = 5; in TEST() local
479 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
485 {n_batch, n_input}, // input tensor in TEST()
487 {n_cell, n_input}, // input_to_input_weight tensor in TEST()
488 {n_cell, n_input}, // input_to_forget_weight tensor in TEST()
489 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
490 {n_cell, n_input}, // input_to_output_weight tensor in TEST()