#include "ONNXHelpers.h"
-#include "mir/ops/PoolOp.h"
+#include "mir/ops/AvgPool2DOp.h"
namespace mir_onnx
{
const std::vector<std::int32_t> padding_after{0, 0};
input = convertONNXToMIR(graph, input);
- auto result =
- createOp<mir::ops::PoolOp>(graph, input, mir::ops::PoolOp::PoolingType::AVG,
- mir::Shape(window_size), mir::Shape(strides), padding_before,
- padding_after, mir::ops::PoolOp::BorderType::ZEROFILLED)
- ->getOutput(0);
+ auto result = createOp<mir::ops::AvgPool2DOp>(graph, input, window_size, strides, padding_before,
+ padding_after, true, mir::DataFormat::NHWC)
+ ->getOutput(0);
result = convertMIRToONNX(graph, result);
context->setNodeOutputs(onnx_node, {result});