summaryrefslogtreecommitdiff
path: root/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h
diff options
context:
space:
mode:
author이한종/동작제어Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>2018-08-14 11:04:07 +0900
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>2018-08-14 11:04:07 +0900
commit3a7967188d0af9f7a4b8eb875cd976fd8e260808 (patch)
treeb7f29c6db0e1a04ef7990f16ee872b50008abac5 /runtimes/neurun/src/kernel/cpu/ReshapeLayer.h
parent4a71c9a720e4aaeb9116f8bf279301bab3175a76 (diff)
downloadnnfw-3a7967188d0af9f7a4b8eb875cd976fd8e260808.tar.gz
nnfw-3a7967188d0af9f7a4b8eb875cd976fd8e260808.tar.bz2
nnfw-3a7967188d0af9f7a4b8eb875cd976fd8e260808.zip
Rename `kernel/cpufallback` to `kernel/cpu` (#2282)
Rename the directory to remove inconsistency with namespace name in the code and backend module(cpu backend is in `backend/cpu`). Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>
Diffstat (limited to 'runtimes/neurun/src/kernel/cpu/ReshapeLayer.h')
-rw-r--r--runtimes/neurun/src/kernel/cpu/ReshapeLayer.h45
1 files changed, 45 insertions, 0 deletions
diff --git a/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h b/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h
new file mode 100644
index 000000000..5a93a0f53
--- /dev/null
+++ b/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h
@@ -0,0 +1,45 @@
+#ifndef __NEURUN_KERNEL_CPU_RESHAPELAYER_H__
+#define __NEURUN_KERNEL_CPU_RESHAPELAYER_H__
+
+#include <NeuralNetworks.h>
+
+#include <arm_compute/runtime/IFunction.h>
+
+#include "internal/Model.h"
+#include "kernel/cpu/OperationUtils.h"
+
+using namespace neurun::kernel::cpu;
+
+namespace neurun
+{
+namespace kernel
+{
+namespace cpu
+{
+
+class ReshapeLayer : public ::arm_compute::IFunction
+{
+public:
+ ReshapeLayer() {}
+
+public:
+ bool reshapeGeneric();
+
+ void configure(uint8_t *inputData, const internal::tflite::operand::Shape &inputShape,
+ uint8_t *outputData, const internal::tflite::operand::Shape &outputShape);
+
+ void run();
+
+private:
+ uint8_t *_inputData;
+ uint8_t *_outputData;
+
+ Shape _inputShape;
+ Shape _outputShape;
+};
+
+} // namespace cpu
+} // namespace kernel
+} // namespace neurun
+
+#endif // __NEURUN_KERNEL_CPU_RESHAPELAYER_H__