Merge branch 'gru_operator' into lstm_operator

src/rewrite_rnn.cpp

@@ -479,11 +479,12 @@ std::vector<instruction_ref> rewrite_rnn::gru_cell(bool is_forward,
 
     instruction_ref hidden_states = prog.end();
     instruction_ref last_output{};
-    long seq_len = static_cast<long>(seq->get_shape().lens()[0]);
-    long hs      = static_cast<long>(r->get_shape().lens()[2]);
+    migraphx::shape seq_shape = seq->get_shape();
+    migraphx::shape r_shape   = r->get_shape();
+    long seq_len              = static_cast<long>(seq_shape.lens()[0]);
+    long hs                   = static_cast<long>(r_shape.lens()[2]);
 
-    migraphx::shape s(seq->get_shape().type(),
-                      {seq->get_shape().lens()[1], static_cast<std::size_t>(hs)});
+    migraphx::shape s(seq_shape.type(), {seq_shape.lens()[1], r_shape.lens()[2]});
     std::vector<int> data(s.elements(), 1);
     auto l1 = prog.add_literal(migraphx::literal{s, data});