#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 <arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h>
#include "internal/arm_compute.h"
#include "internal/arm_compute/Cast.h"
builder.append("FullyConnected", std::move(fn));
}
- else
- throw std::runtime_error("Not supported, yet");
+ else // NEON
+ {
+ auto fn = nnfw::make_unique<::arm_compute::NEFullyConnectedLayer>();
+
+ fn->configure(input_alloc, weight_alloc, bias_alloc, output_alloc);
+
+ builder.append("FullyConnected", std::move(fn));
+ }
ActivationBuilder{builder}.append(param.activation, output_alloc);
};