diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-05-08 12:01:57 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:51:50 +0000 |
commit | 750641dd6aab1e5e62d1875b97b230312bb87959 (patch) | |
tree | b3b180c07d7769cb32a6f35b6d0df2384a4638b0 /arm_compute/core/utils/misc/ShapeCalculator.h | |
parent | aa3240d3e2a575c436ec60ea0a31e8375d997425 (diff) | |
download | ComputeLibrary-750641dd6aab1e5e62d1875b97b230312bb87959.tar.gz |
COMPMID-1052 - Rework validate method in CLGEMM
Change-Id: Iece5bd6478b5fac5164abff30c1e63e8a77291a9
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/130374
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/utils/misc/ShapeCalculator.h')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index 9543d989b8..30d3f9bb62 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -358,6 +358,14 @@ inline TensorShape compute_rnn_shape(const ITensorInfo *input, const unsigned in return output_shape; } +inline TensorShape compute_mm_shape(const ITensorInfo &input0, const ITensorInfo &input1, bool is_interleaved_transposed, const GEMMReshapeInfo &reshape_info) +{ + TensorShape tensor_shape{ input0.tensor_shape() }; + tensor_shape.set(0, is_interleaved_transposed ? reshape_info.n() : input1.dimension(0)); + tensor_shape.set(1, is_interleaved_transposed ? reshape_info.m() : input0.dimension(1)); + + return tensor_shape; +} } // namespace shape_calculator } // namespace misc } // namespace arm_compute |