diff options
author | thecha01 <theo.charalambous@arm.com> | 2020-08-12 17:25:51 +0100 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-09-04 15:55:01 +0000 |
commit | 2bfadd92d638fd4e7b8b3398ce95f15930749189 (patch) | |
tree | 7c4a85890884c468e1e736353577e10ea18bc928 /arm_compute/core | |
parent | 0a05e6a051e6d9028131de10f00b80bbedf44097 (diff) | |
download | ComputeLibrary-2bfadd92d638fd4e7b8b3398ce95f15930749189.tar.gz |
Add StridedSliceLayerNode to graph API
Signed-off-by: thecha01 <theo.charalambous@arm.com>
Change-Id: I6d6fb2b053c74e35a86841621486bc0cd34b12b3
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3911
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/Types.h | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 022e20dadc..c0e0683fe6 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -1690,6 +1690,44 @@ private: bool _is_scaled; }; +class StridedSliceLayerInfo +{ +public: + /** Default Constructor + * + * @param[in] begin_mask (Optional) If the ith bit of begin_mask is set, starts[i] is ignored and the fullest possible range in that dimension is used instead. + * @param[in] end_mask (Optional) If the ith bit of end_mask is set, ends[i] is ignored and the fullest possible range in that dimension is used instead. + * @param[in] shrink_axis_mask (Optional) If the ith bit of shrink_axis_mask is set, it implies that the ith specification shrinks the dimensionality by 1. + */ + StridedSliceLayerInfo(int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0) + : _begin_mask(begin_mask), _end_mask(end_mask), _shrink_axis_mask(shrink_axis_mask) + { + } + + /* Get the begin mask value */ + int32_t begin_mask() const + { + return _begin_mask; + } + + /* Get the end mask value */ + int32_t end_mask() const + { + return _end_mask; + } + + /* Get the shrink axis mask value */ + int32_t shrink_axis_mask() const + { + return _shrink_axis_mask; + } + +private: + int32_t _begin_mask; + int32_t _end_mask; + int32_t _shrink_axis_mask; +}; + /** Convolution Layer Weights Information class. This class stores the necessary information to compute convolution layer when the weights are already reshaped */ class WeightsInfo { |