diff --git a/unit_tests/operator/Test_MetaOperator.cpp b/unit_tests/operator/Test_MetaOperator.cpp index 5fc68acf157400728740eb10dc6c1ae962164df7..c0e9be1c6062eaf311d5eaf2515df2b4fd2b8a9e 100644 --- a/unit_tests/operator/Test_MetaOperator.cpp +++ b/unit_tests/operator/Test_MetaOperator.cpp @@ -304,10 +304,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { auto microGraphScheduler = std::dynamic_pointer_cast<MetaOperator_Op>(op)->getMicroGraphScheduler(); microGraphScheduler->saveSchedulingDiagram("lstm_values_scheduling"); - op->getOutput(1)->print(); + op->getOutput(0)->print(); myHiddenState->print(); - REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState)); + REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState)); } SECTION("LSTM(forward_values_seq)") { auto pop = Pop(); @@ -360,10 +360,10 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { myGraph->save("lstm_seq_mygraph", true, true); - op->getOutput(1)->print(); + op->getOutput(0)->print(); myHiddenState->print(); - REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState)); + REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState)); } SECTION("LSTM(forward_values_seq_flatten)") { auto pop = Pop(); @@ -421,9 +421,9 @@ TEST_CASE("[cpu/operator] MetaOperator", "[MetaOperator][CPU]") { scheduler.forward(true, true); scheduler.saveSchedulingDiagram("lstm_seq_flatten_schedule"); - op->getOutput(1)->print(); + op->getOutput(0)->print(); myHiddenState->print(); - REQUIRE(approxEq<float>(*(op->getOutput(1)), *myHiddenState)); + REQUIRE(approxEq<float>(*(op->getOutput(0)), *myHiddenState)); } } \ No newline at end of file