Lines Matching refs:n_cell
78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel() argument
95 input_shapes.push_back({n_batch, n_cell}); in LSTMOpModel()
117 {n_batch, n_cell * (use_cifg ? 3 : 4)}, in LSTMOpModel()
119 {n_batch, n_cell}, in LSTMOpModel()
139 CellStateIn_.insert(CellStateIn_.end(), n_batch * n_cell, 0.f); in LSTMOpModel()
276 const int n_cell = 4; in TEST() local
279 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, 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()
292 {n_cell, n_output}, // recurrent_to_input_weight tensor in TEST()
293 {n_cell, n_output}, // recurrent_to_forget_weight tensor in TEST()
294 {n_cell, n_output}, // recurrent_to_cell_weight tensor in TEST()
295 {n_cell, n_output}, // recurrent_to_output_weight tensor in TEST()
301 {n_cell}, // input_gate_bias tensor in TEST()
302 {n_cell}, // forget_gate_bias tensor in TEST()
303 {n_cell}, // cell_bias tensor in TEST()
304 {n_cell}, // output_gate_bias tensor in TEST()
380 const int n_cell = 4; in TEST() local
383 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, 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()
397 {n_cell, n_output}, // recurrent_to_forget_weight tensor in TEST()
398 {n_cell, n_output}, // recurrent_to_cell_weight tensor in TEST()
399 {n_cell, n_output}, // recurrent_to_output_weight tensor in TEST()
402 {n_cell}, // cell_to_forget_weight tensor in TEST()
403 {n_cell}, // cell_to_output_weight tensor in TEST()
405 {n_cell}, // input_gate_bias tensor in TEST()
406 {n_cell}, // forget_gate_bias tensor in TEST()
407 {n_cell}, // cell_bias tensor in TEST()
408 {n_cell}, // output_gate_bias tensor in TEST()
476 const int n_cell = 20; in TEST() local
479 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, 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()
492 {n_cell, n_output}, // recurrent_to_input_weight tensor in TEST()
493 {n_cell, n_output}, // recurrent_to_forget_weight tensor in TEST()
494 {n_cell, n_output}, // recurrent_to_cell_weight tensor in TEST()
495 {n_cell, n_output}, // recurrent_to_output_weight tensor in TEST()
497 {n_cell}, // cell_to_input_weight tensor in TEST()
498 {n_cell}, // cell_to_forget_weight tensor in TEST()
499 {n_cell}, // cell_to_output_weight tensor in TEST()
501 {n_cell}, // input_gate_bias tensor in TEST()
502 {n_cell}, // forget_gate_bias tensor in TEST()
503 {n_cell}, // cell_bias tensor in TEST()
504 {n_cell}, // output_gate_bias tensor in TEST()
506 {n_output, n_cell}, // projection_weight tensor in TEST()