diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2020-08-18 10:07:35 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-08-20 10:17:54 +0000 |
commit | a3e6b6d2c556227aea0c145014612beb6e5c15c8 (patch) | |
tree | 3417d797333be62d2bff357660c7a6c3f8df10e6 /src/core | |
parent | 0bc80daf319ea3219ca6a6fa200118dc859ee460 (diff) | |
download | ComputeLibrary-a3e6b6d2c556227aea0c145014612beb6e5c15c8.tar.gz |
COMPMID-3696: Padded dilated Conv2D segmentation
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I8045166d2d3202612fec3f6bf9651f3e6bfcb20f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3783
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-by: Aleksandr Nikolaev <aleksandr.nikolaev@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core')
-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 0971d2aa73..c1a1c2ecf0 100644 --- a/src/core/TensorInfo.cpp +++ b/src/core/TensorInfo.cpp @@ -268,7 +268,7 @@ std::tuple<Strides, size_t, size_t> TensorInfo::calculate_padding_requirements(c const unsigned int idx_last_dimension = _tensor_shape.num_dimensions() - 1; - required_total_size = _tensor_shape[idx_last_dimension] * required_strides[idx_last_dimension]; + required_total_size = static_cast<size_t>(_tensor_shape[idx_last_dimension]) * required_strides[idx_last_dimension]; break; } } @@ -360,7 +360,7 @@ ITensorInfo &TensorInfo::set_tensor_shape(const TensorShape &shape) else { const unsigned int idx_last_dimension = _tensor_shape.num_dimensions() - 1; - _total_size = _tensor_shape[idx_last_dimension] * _strides_in_bytes[idx_last_dimension]; + _total_size = static_cast<size_t>(_tensor_shape[idx_last_dimension]) * _strides_in_bytes[idx_last_dimension]; } std::tie(_strides_in_bytes, _offset_first_element_in_bytes, _total_size) = calculate_padding_requirements(_padding); |