diff options
Diffstat (limited to 'src/core/SubTensorInfo.cpp')
-rw-r--r-- | src/core/SubTensorInfo.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/SubTensorInfo.cpp b/src/core/SubTensorInfo.cpp index 237f1333f2..2db76b475a 100644 --- a/src/core/SubTensorInfo.cpp +++ b/src/core/SubTensorInfo.cpp @@ -123,11 +123,11 @@ bool SubTensorInfo::extend_padding(const PaddingSize &padding) return _parent->extend_padding(padding); } -size_t SubTensorInfo::offset_element_in_bytes(const Coordinates &pos) const +int32_t SubTensorInfo::offset_element_in_bytes(const Coordinates &pos) const { ARM_COMPUTE_ERROR_ON_COORDINATES_DIMENSIONS_GTE(pos, _tensor_shape.num_dimensions()); - size_t offset = offset_first_element_in_bytes(); + int32_t offset = offset_first_element_in_bytes(); const Strides &strides = strides_in_bytes(); for(size_t i = 0; i < _tensor_shape.num_dimensions(); ++i) |