aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph2/backends/NEON/NESubTensorHandle.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2018-03-08 16:01:29 +0000
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:49:16 +0000
commitee33ea5a6e1aa0faac1cc8b5a269bd4f89854821 (patch)
tree0baf159ae4a61d07cc765ad6bb1a2fb42c403081 /arm_compute/graph2/backends/NEON/NESubTensorHandle.h
parente86a09fe4c5aa9037787e13ee55cba2b049d5ea5 (diff)
downloadComputeLibrary-ee33ea5a6e1aa0faac1cc8b5a269bd4f89854821.tar.gz
COMPMID-996: Add support for grouped convolution.
Change-Id: I279e29ce20b3dde57445264dc11491f127b44d70 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124429 Reviewed-by: Anthony Barbier <anthony.barbier@arm.com> Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph2/backends/NEON/NESubTensorHandle.h')
-rw-r--r--arm_compute/graph2/backends/NEON/NESubTensorHandle.h3
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/graph2/backends/NEON/NESubTensorHandle.h b/arm_compute/graph2/backends/NEON/NESubTensorHandle.h
index e027b0cc56..eacdfe0fb4 100644
--- a/arm_compute/graph2/backends/NEON/NESubTensorHandle.h
+++ b/arm_compute/graph2/backends/NEON/NESubTensorHandle.h
@@ -43,8 +43,9 @@ public:
* @param[in] parent_handle Parent tensor handle
* @param[in] shape Sub-Tensor shape
* @param[in] coords Starting coordinates
+ * @param[in] extend_parent Extends parent shape if true
*/
- NESubTensorHandle(ITensorHandle *parent_handle, const TensorShape &shape, const Coordinates &coords);
+ NESubTensorHandle(ITensorHandle *parent_handle, const TensorShape &shape, const Coordinates &coords, bool extend_parent = false);
/** Destructor: free the tensor's memory */
~NESubTensorHandle() = default;
/** Allow instances of this class to be move constructed */