Lines Matching full:n_batch
76 LSTMOpModel(uint32_t n_batch, uint32_t n_input,
91 input_shapes.push_back({n_batch, n_output});
92 input_shapes.push_back({n_batch, n_cell});
114 {n_batch, n_cell * (use_cifg ? 3 : 4)},
115 {n_batch, n_output},
116 {n_batch, n_cell},
117 {n_batch, n_output},
134 Input_.insert(Input_.end(), n_batch * n_input, 0.f);
135 OutputStateIn_.insert(OutputStateIn_.end(), n_batch * n_output, 0.f);
136 CellStateIn_.insert(CellStateIn_.end(), n_batch * n_cell, 0.f);
274 const int n_batch = 1;
280 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
286 {n_batch, n_input}, // input tensor
385 const int n_batch = 1;
391 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
397 {n_batch, n_input}, // input tensor
490 const int n_batch = 2;
495 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
501 {n_batch, n_input}, // input tensor
