#include <arm_compute/runtime/NEON/functions/NEPoolingLayer.h>
#include <arm_compute/runtime/NEON/functions/NEActivationLayer.h>
#include <arm_compute/runtime/NEON/functions/NEConvolutionLayer.h>
+#include <arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h>
#include "internal/arm_compute.h"
#include "internal/arm_compute/Cast.h"
builder.append("DepthwiseConv2D", std::move(fn));
}
else
- throw std::runtime_error("Not supported, yet");
+ {
+ auto fn = nnfw::make_unique<::arm_compute::NEDepthwiseConvolutionLayer>();
+
+ fn->configure(ifm_alloc, ker_alloc, bias_alloc, ofm_alloc, conv_info, param.multipler);
+
+ builder.append("DepthwiseConv2D", std::move(fn));
+ }
ActivationBuilder{builder}.append(param.activation, ofm_alloc);
};
builder.append("Reshape", std::move(fn));
}
else
- throw std::runtime_error("Not supported, yet");
+ {
+ auto fn = nnfw::make_unique<GenericReshapeLayer>();
+
+ fn->configure(input_alloc, output_alloc);
+
+ builder.append("Reshape", std::move(fn));
+ }
};
_builder.addStage(stage);