diff options
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 |
commit | 3a7967188d0af9f7a4b8eb875cd976fd8e260808 (patch) | |
tree | b7f29c6db0e1a04ef7990f16ee872b50008abac5 /runtimes/neurun/src/kernel/cpu/ReshapeLayer.h | |
parent | 4a71c9a720e4aaeb9116f8bf279301bab3175a76 (diff) | |
download | nnfw-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.h | 45 |
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__ |