diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-11-29 11:06:49 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:41:58 +0000 |
| commit | 45bcc3a1c287a208098ae99288273a5129ddd5eb (patch) | |
| tree | f4f957dbc76f8e8e9a4871b16652e1033bcd4c73 /arm_compute/core/Dimensions.h | |
| parent | 303be90ee1f03f75309b421297ba16428ea98ea5 (diff) | |
| download | ComputeLibrary-45bcc3a1c287a208098ae99288273a5129ddd5eb.tar.gz | |
COMPMID-661: QASYMM8 support for fully connected layer.
Change-Id: I70e04d3a175ba366432ada98e9ca893c9f81b260
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/111094
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/Dimensions.h')
| -rw-r--r-- | arm_compute/core/Dimensions.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/arm_compute/core/Dimensions.h b/arm_compute/core/Dimensions.h index 3d9a3fa7ff..912b9d57d7 100644 --- a/arm_compute/core/Dimensions.h +++ b/arm_compute/core/Dimensions.h @@ -141,6 +141,17 @@ public: std::fill(_id.begin() + _num_dimensions, _id.end(), 0); } + /** Collapse dimensions starting from a given point + * + * @param[in] start Starting point of collapsing dimensions + */ + void collapse_from(size_t start) + { + ARM_COMPUTE_ERROR_ON(start > num_dimensions()); + + collapse(num_dimensions() - start, start); + } + /** Returns a read/write iterator that points to the first element in the dimension array. */ typename std::array<T, num_max_dimensions>::iterator begin() { |