diff options
Diffstat (limited to 'src/core/TensorInfo.cpp')
-rw-r--r-- | src/core/TensorInfo.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/TensorInfo.cpp b/src/core/TensorInfo.cpp index 33d682f772..cd36e8be2c 100644 --- a/src/core/TensorInfo.cpp +++ b/src/core/TensorInfo.cpp @@ -384,11 +384,11 @@ ITensorInfo &TensorInfo::reset_padding() return *this; } -size_t TensorInfo::offset_element_in_bytes(const Coordinates &pos) const +int32_t TensorInfo::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; for(size_t i = 0; i < _tensor_shape.num_dimensions(); ++i) { |