aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEStackLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEStackLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEStackLayer.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEStackLayer.h b/arm_compute/runtime/NEON/functions/NEStackLayer.h
index 6032dae0cb..ba262371cd 100644
--- a/arm_compute/runtime/NEON/functions/NEStackLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEStackLayer.h
@@ -73,9 +73,9 @@ public:
void run() override;
private:
- std::vector<ITensor *> _input;
- std::unique_ptr<NEStackLayerKernel[]> _stack_kernels;
- unsigned int _num_inputs;
+ std::vector<ITensor *> _input;
+ std::vector<NEStackLayerKernel> _stack_kernels;
+ unsigned int _num_inputs;
};
} // namespace arm_compute
#endif /* __ARM_COMPUTE_NESTACKLAYER_H__ */