aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/utils
diff options
context:
space:
mode:
authorgiuros01 <giuseppe.rossini@arm.com>2018-11-21 09:59:17 +0000
committerGiuseppe Rossini <giuseppe.rossini@arm.com>2018-11-27 20:22:05 +0000
commit3175fcf63249673f33fd1638879adad4baab545b (patch)
tree5b870cd5180c122eaa475bf6a08bb3cb8228e09e /arm_compute/core/utils
parent0a80d92bcdbdc4f3974d0fbcaa529a71ed539559 (diff)
downloadComputeLibrary-3175fcf63249673f33fd1638879adad4baab545b.tar.gz
COMPMID-1720: CL: Implement Tile
Change-Id: I2a18f0acea382960a8bc71a8f56928a5998f0dd6
Diffstat (limited to 'arm_compute/core/utils')
-rw-r--r--arm_compute/core/utils/misc/ShapeCalculator.h10
1 files changed, 10 insertions, 0 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h
index da9ff56fd0..38906dfc9b 100644
--- a/arm_compute/core/utils/misc/ShapeCalculator.h
+++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -671,6 +671,16 @@ inline TensorShape compute_padded_shape(const TensorShape &input_shape, const Pa
return padded_shape;
}
+inline TensorShape compute_tiled_shape(const TensorShape &input_shape, const Multiples &multiples)
+{
+ TensorShape tiled_shape = input_shape;
+ for(size_t dim = 0; dim < multiples.size(); ++dim)
+ {
+ tiled_shape.set(dim, input_shape[dim] * multiples[dim]);
+ }
+ return tiled_shape;
+}
+
inline TensorShape compute_upsample_shape(const ITensorInfo &input, const Size2D &info)
{
const DataLayout data_layout = input.data_layout();