From 917959c88361e8148696c156453f69c6ae0c95c0 Mon Sep 17 00:00:00 2001 From: John Kesapides Date: Mon, 4 Feb 2019 12:37:29 +0000 Subject: COMPMID-1281 Investigate concatenation for RNN/LSTM NEON Change-Id: I7f099348a361a6f2d4efb30618f58bd44dd41e6c Signed-off-by: John Kesapides Reviewed-on: https://review.mlplatform.org/c/712 Reviewed-by: Giuseppe Rossini Tested-by: Arm Jenkins --- arm_compute/core/utils/misc/ShapeCalculator.h | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'arm_compute/core/utils') diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index b256e73146..9d36405041 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -1162,6 +1162,11 @@ inline TensorShape extract_shape(T *data) return data->info()->tensor_shape(); } +inline TensorShape extract_shape(const ITensorInfo *data) +{ + return data->tensor_shape(); +} + inline TensorShape extract_shape(ITensorInfo *data) { return data->tensor_shape(); -- cgit v1.2.1