#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 <arm_compute/runtime/NEON/functions/NEFloor.h>
#include "internal/arm_compute.h"
#include "internal/arm_compute/Cast.h"
builder.append("Floor", std::move(fn));
}
else
- throw std::runtime_error("Not supported, yet");
+ {
+ auto fn = nnfw::make_unique<::arm_compute::NEFloor>();
+
+ fn->configure(ifm_alloc, ofm_alloc);
+
+ builder.append("Floor", std::move(fn));
+ }
};
_builder.addStage(stage);