diff options
author | John Kesapides <john.kesapides@arm.com> | 2019-02-04 12:37:29 +0000 |
---|---|---|
committer | Pablo Marquez <pablo.tello@arm.com> | 2019-03-14 09:29:15 +0000 |
commit | 917959c88361e8148696c156453f69c6ae0c95c0 (patch) | |
tree | fbdadd65904b446edcc275f9e85874d2dfb7f8d3 /arm_compute/core | |
parent | 894066de8cc26d1a3aca62dcaa6b30a2a1116028 (diff) | |
download | ComputeLibrary-917959c88361e8148696c156453f69c6ae0c95c0.tar.gz |
COMPMID-1281 Investigate concatenation for RNN/LSTM NEON
Change-Id: I7f099348a361a6f2d4efb30618f58bd44dd41e6c
Signed-off-by: John Kesapides <john.kesapides@arm.com>
Reviewed-on: https://review.mlplatform.org/c/712
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 5 |
1 files changed, 5 insertions, 0 deletions
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(); |