));     EXPECT_EQ(lstm_cell->get_output_element_type(0), element::f00);     EXPECT_EQ(lstm_cell->get_output_shape(0), (Shape{batch_size, hidden_size}));     EXPECT_EQ(lstm_cell->get_output_element_type(0), element::f00);     EXPECT_EQ(lstm_cell->get_output_shape(0), (Shape{batch_size, hidden_size})); 