#include "ONNXHelpers.h"
#include "AttributeHelpers.h"
-#include "mir/ops/ReduceOp.h"
+#include "mir/ops/ReduceMeanOp.h"
#include <numeric>
bool keep_dims = static_cast<bool>(keepdims);
mir::Graph *graph = context->getGraph();
- auto result = createOp<mir::ops::ReduceOp>(graph, inputs[0], reduce_dims, keep_dims,
- mir::ops::ReduceOp::FuncType::mean)
- ->getOutput(0);
+ auto result =
+ createOp<mir::ops::ReduceMeanOp>(graph, inputs[0], reduce_dims, keep_dims)->getOutput(0);
context->setNodeOutputs(onnx_node, {result});
}