diff options
Diffstat (limited to 'arm_compute/runtime/SubTensor.h')
| -rw-r--r-- | arm_compute/runtime/SubTensor.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/arm_compute/runtime/SubTensor.h b/arm_compute/runtime/SubTensor.h index bdb229de49..ba2f8682af 100644 --- a/arm_compute/runtime/SubTensor.h +++ b/arm_compute/runtime/SubTensor.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -39,11 +39,12 @@ class SubTensor : public ITensor public: /** Constructor * - * @param[in] parent Parent tensor - * @param[in] tensor_shape Shape of the subtensor - * @param[in] coords Coordinates of the first subtensor element inside the parent tensor. + * @param[in] parent Parent tensor + * @param[in] tensor_shape Shape of the subtensor + * @param[in] coords Coordinates of the first subtensor element inside the parent tensor. + * @param[in] extend_parent (Optional) Extend parent with subtensor shape if subtensor indexes out of bounds */ - SubTensor(ITensor *parent, const TensorShape &tensor_shape, const Coordinates &coords); + SubTensor(ITensor *parent, const TensorShape &tensor_shape, const Coordinates &coords, bool extend_parent = false); /** Destructor: free the tensor's memory */ ~SubTensor() = default; /** Restrict instances of this class to be copy constructed */ |