diff options
Diffstat (limited to 'src/core/TensorInfo.cpp')
-rw-r--r-- | src/core/TensorInfo.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/core/TensorInfo.cpp b/src/core/TensorInfo.cpp index 2008217c85..24988e2217 100644 --- a/src/core/TensorInfo.cpp +++ b/src/core/TensorInfo.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016, 2018 ARM Limited. + * Copyright (c) 2016-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -364,6 +364,8 @@ ITensorInfo &TensorInfo::set_tensor_shape(TensorShape shape) _total_size = _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); + _valid_region = ValidRegion{ Coordinates(), _tensor_shape }; return *this; } |