diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEReorderLayer.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEReorderLayer.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEReorderLayer.cpp b/src/runtime/NEON/functions/NEReorderLayer.cpp index f9bbeba90c..53a7718902 100644 --- a/src/runtime/NEON/functions/NEReorderLayer.cpp +++ b/src/runtime/NEON/functions/NEReorderLayer.cpp @@ -21,6 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#if defined(__aarch64__) + #include "arm_compute/runtime/NEON/NEScheduler.h" #include "arm_compute/runtime/NEON/functions/NEReorderLayer.h" @@ -50,4 +52,6 @@ Status NEReorderLayer::validate(const ITensorInfo *input, const ITensorInfo *out return NEReorderKernel::validate(input, output, input_wf, output_wf); } -} // namespace arm_compute
\ No newline at end of file +} // namespace arm_compute + +#endif // defined(__aarch64__)
\ No newline at end of file |