diff options
Diffstat (limited to 'arm_compute/graph/Tensor.h')
| -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 dcb0c661d6..94822c2c1e 100644 --- a/arm_compute/graph/Tensor.h +++ b/arm_compute/graph/Tensor.h @@ -68,6 +68,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) { |