aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonDequantizeWorkload.hpp
diff options
context:
space:
mode:
authorMatthew Bentham <matthew.bentham@arm.com>2020-01-24 23:11:43 +0000
committerNikhil Raj Arm <nikhil.raj@arm.com>2020-02-04 15:42:53 +0000
commit5e98b012597c13dc0acb048bd10a84a2a397a346 (patch)
treedd4ee3aa279277e328a4a01905b23605dd679923 /src/backends/neon/workloads/NeonDequantizeWorkload.hpp
parentaa920c56838c2a0b31bd4e3c54bd57ff2f20969e (diff)
downloadarmnn-5e98b012597c13dc0acb048bd10a84a2a397a346.tar.gz
Clean up header usage a bit in NEON backend
Including NEFunctions.h is unnecessary and adds about a second to compile time each translation unit in which it appears, so we should use just the header file with the arm compute function declarations that we need. Signed-off-by: Matthew Bentham <matthew.bentham@arm.com> Change-Id: I605d0eb82ccf2aafa35381a5d9d54337d3fe17a7
Diffstat (limited to 'src/backends/neon/workloads/NeonDequantizeWorkload.hpp')
-rw-r--r--src/backends/neon/workloads/NeonDequantizeWorkload.hpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonDequantizeWorkload.hpp b/src/backends/neon/workloads/NeonDequantizeWorkload.hpp
index a6616376e7..63ea783474 100644
--- a/src/backends/neon/workloads/NeonDequantizeWorkload.hpp
+++ b/src/backends/neon/workloads/NeonDequantizeWorkload.hpp
@@ -7,7 +7,8 @@
#include <backendsCommon/Workload.hpp>
-#include <arm_compute/runtime/NEON/NEFunctions.h>
+#include <arm_compute/core/Error.h>
+#include <arm_compute/runtime/IFunction.h>
#include <functional>
@@ -25,7 +26,7 @@ public:
void Execute() const override;
private:
- mutable std::unique_ptr<arm_compute::NEDequantizationLayer> m_Layer;
+ mutable std::unique_ptr<arm_compute::IFunction> m_Layer;
};
} //namespace armnn