src: ocl: rnn: fix wrong gemm dimension

oneapi-src · Aug 12, 2019 · eb3c866 · eb3c866
1 parent f6e4b97
commit eb3c866
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ocl/ref_rnn.hpp b/src/ocl/ref_rnn.hpp
@@ -261,7 +261,7 @@ struct _ref_rnn_common_t : public primitive_t {
                     create_gemm_pd(&gemm_iter_pd_, sic, batch, n_gates * dic,
                         slc, n_gates * dic, wic, weights_type, src_type,
                         src_type, false, 0.0f),
-                    create_gemm_pd(&gemm_layer_pd_, sic, batch, n_gates * dic,
+                    create_gemm_pd(&gemm_layer_pd_, slc, batch, n_gates * dic,
                         sic, n_gates * dic, wic, weights_type, src_type,
                         src_type, false, 0.0f),
                     create_gemm_pd(&gemm_diff_wei_layer_pd_, n_gates * dic, slc,