Skip to content

Commit 5824efc

Browse files
committed
update types for lstm init state oneAPI
1 parent dde4a4d commit 5824efc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

hls4ml/templates/oneapi/firmware/nnet_utils/nnet_recurrent.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -611,10 +611,10 @@ void lstm_init_state(const data_T &data, const h_T &hidden_state_initial, const
611611

612612
[[intel::fpga_register]] h_T hidden_state[CONFIG_T::n_timesteps + 1];
613613
[[intel::fpga_register]] h_T hidden_state_temp;
614-
[[intel::fpga_register]] h_T cell_state[CONFIG_T::n_timesteps + 1];
615-
[[intel::fpga_register]] h_T cell_state_temp; // should this be updated to a differnt type
614+
[[intel::fpga_register]] hc_T cell_state[CONFIG_T::n_timesteps + 1];
615+
[[intel::fpga_register]] hc_T cell_state_temp; // should this be updated to a differnt type
616616
[[intel::fpga_register]] h_T h;
617-
[[intel::fpga_register]] h_T c;
617+
[[intel::fpga_register]] hc_T c;
618618
[[intel::fpga_register]] in_T in;
619619

620620
// Set initially hidden state (output) to zero

0 commit comments

Comments
 (0)