Skip to content

Commit cf9c726

Browse files
committed
fix simplernn in oneAPI
1 parent 46b590f commit cf9c726

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ void simple_rnn(const data_T &data, res_T &res, const typename CONFIG_T::weight_
336336
// Write result
337337
#pragma unroll
338338
for (int x = 0; x < CONFIG_T::n_out; x++) {
339-
hidden_state[x][i + 1] = h[x];
339+
hidden_state[i + 1][x] = h[x];
340340
}
341341
}
342342

0 commit comments

Comments
 (0)