diff --git a/unit_tests/operator/Test_ReduceMeanImpl.cpp b/unit_tests/operator/Test_ReduceMeanImpl.cpp index 58ed69e75726d43c7de7ebb987f00263cc28b905..0aed151516fd38251d95c6615696858d7fb17da6 100644 --- a/unit_tests/operator/Test_ReduceMeanImpl.cpp +++ b/unit_tests/operator/Test_ReduceMeanImpl.cpp @@ -207,12 +207,12 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)", "[ReduceMean][CPU]") { } }); - std::shared_ptr<Node> myReduceSum = ReduceSum({}, 0, 1); - auto op = std::static_pointer_cast<OperatorTensor>(myReduceSum -> getOperator()); + std::shared_ptr<Node> myReduceMean = ReduceMean({}, 0, 1); + auto op = std::static_pointer_cast<OperatorTensor>(myReduceMean -> getOperator()); op->associateInput(0,myInput); op->setDataType(DataType::Float32); op->setBackend("cpu"); - myReduceSum->forward(); + myReduceMean->forward(); op->getOutput(0)->print(); REQUIRE(*(op->getOutput(0)) == *myInput);