diff options
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/CL/CLSubTensor.h | 11 | ||||
-rw-r--r-- | arm_compute/runtime/SubTensor.h | 11 |
2 files changed, 12 insertions, 10 deletions
diff --git a/arm_compute/runtime/CL/CLSubTensor.h b/arm_compute/runtime/CL/CLSubTensor.h index 4bab164779..b6e9a29df4 100644 --- a/arm_compute/runtime/CL/CLSubTensor.h +++ b/arm_compute/runtime/CL/CLSubTensor.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 CLSubTensor : public ICLTensor 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 */ - CLSubTensor(ICLTensor *parent, const TensorShape &tensor_shape, const Coordinates &coords); + CLSubTensor(ICLTensor *parent, const TensorShape &tensor_shape, const Coordinates &coords, bool extend_parent = false); /** Destructor: free the tensor's memory */ ~CLSubTensor() = default; /** Restrict instances of this class to be copy constructed */ 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 */ |