diff options
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/graph/Tensor.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/arm_compute/graph/Tensor.h b/arm_compute/graph/Tensor.h index 9fdd56db6..dbe2ba595 100644 --- a/arm_compute/graph/Tensor.h +++ b/arm_compute/graph/Tensor.h @@ -67,6 +67,16 @@ public: * @param[in] accessor Tensor accessor */ template <typename AccessorType> + Tensor(TensorInfo &&info, std::unique_ptr<AccessorType> &&accessor) + : _target(TargetHint::DONT_CARE), _info(info), _accessor(std::move(accessor)), _tensor(nullptr) + { + } + /** Constructor + * + * @param[in] info Tensor info to use + * @param[in] accessor Tensor accessor + */ + template <typename AccessorType> Tensor(TensorInfo &&info, AccessorType &&accessor) : _target(TargetHint::DONT_CARE), _info(info), _accessor(arm_compute::support::cpp14::make_unique<AccessorType>(std::forward<AccessorType>(accessor))), _tensor(nullptr) { |