#include <arm_compute/runtime/NEON/functions/NEArithmeticAddition.h>
#include <arm_compute/runtime/NEON/functions/NEArithmeticSubtraction.h>
#include <arm_compute/runtime/NEON/functions/NEPixelWiseMultiplication.h>
+#include <arm_compute/runtime/NEON/functions/NEPoolingLayer.h>
#include <arm_compute/runtime/NEON/functions/NEActivationLayer.h>
#include "internal/arm_compute.h"
builder.append("MaxPool2D", std::move(fn));
}
else
- throw std::runtime_error("Not supported, yet");
+ {
+ std::unique_ptr<::arm_compute::NEPoolingLayer> fn{new ::arm_compute::NEPoolingLayer};
+
+ fn->configure(ifm_alloc, ofm_alloc, info);
+
+ builder.append("MaxPool2D", std::move(fn));
+ }
ActivationBuilder{builder}.append(param.activation, ofm_alloc);
};
builder.append("AvgPool2D", std::move(fn));
}
else
- throw std::runtime_error("Not supported, yet");
+ {
+ std::unique_ptr<::arm_compute::NEPoolingLayer> fn{new ::arm_compute::NEPoolingLayer};
+
+ fn->configure(ifm_alloc, ofm_alloc, info);
+
+ builder.append("AvgPool2D", std::move(fn));
+ }
ActivationBuilder{builder}.append(param.activation, ofm_alloc);
};