Skip to content

Commit

Permalink
fix compilation on windows
Browse files Browse the repository at this point in the history
  • Loading branch information
michal-miotk committed Nov 24, 2024
1 parent 0a9756d commit 2a068c2
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ struct lstm_seq_onednn : typed_primitive_onednn_impl<lstm_seq> {

std::unordered_map<int, dnnl::memory> get_arguments(lstm_seq_inst& instance) const override {
std::unordered_map<int, dnnl::memory> args;
std::vector<std::vector<uint>> dnnl_arg{{DNNL_ARG_SRC_LAYER, DNNL_ARG_SRC_ITER, DNNL_ARG_SRC_ITER_C}, {DNNL_ARG_WEIGHTS_LAYER, DNNL_ARG_WEIGHTS_ITER,
std::vector<std::vector<unsigned int>> dnnl_arg{{DNNL_ARG_SRC_LAYER, DNNL_ARG_SRC_ITER, DNNL_ARG_SRC_ITER_C}, {DNNL_ARG_WEIGHTS_LAYER, DNNL_ARG_WEIGHTS_ITER,
DNNL_ARG_BIAS}, {DNNL_ARG_DST_LAYER, DNNL_ARG_DST_ITER, DNNL_ARG_DST_ITER_C}};

for (int i = 0; i < 3; i++) {
Expand Down

0 comments on commit 2a068c2

Please sign in to comment.