diff options
author | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-30 17:20:26 +0000 |
---|---|---|
committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-12-03 11:48:39 +0000 |
commit | 08a4517905da959b6e3401cc24f5e2018f9b51ac (patch) | |
tree | 9f6e2343ff22a53526a2f04e1ad22ecd91262767 /arm_compute | |
parent | 164a2727d3bbce0e575d24b7db787c85e2e2c203 (diff) | |
download | ComputeLibrary-08a4517905da959b6e3401cc24f5e2018f9b51ac.tar.gz |
COMPMID-1497: Use reshaped B strategy in GEMM by default
NELSTM, NEFullyConnectedLayer(For quantised types only), NERNN and NEWinogradLayer were all defaulting to on-the-fly reshaping of B
Fixed a bug in GemmInterleaved: it was ignoring the 'multis' dimension of the tensor to allocate the memory for B reshaped
Change-Id: I7b30f7f57fc65d6a03cccde0bf5515a811f17b54
Reviewed-on: https://review.mlplatform.org/323
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/Types.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 7d632fec28..d46c93247c 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -1646,8 +1646,8 @@ class GEMMInfo public: /** Default constructor */ GEMMInfo() - : _is_a_reshaped(false), _is_b_reshaped(false), _reshape_b_only_on_first_run(false), _depth_output_gemm3d(0), _reinterpret_input_as_3d(false), _retain_internal_weights(false), - _gemmlowp_output_stage(), _fp_mixed_precision(false) + : _is_a_reshaped(false), _is_b_reshaped(false), _reshape_b_only_on_first_run(true), _depth_output_gemm3d(0), _reinterpret_input_as_3d(false), _retain_internal_weights(false), _gemmlowp_output_stage(), + _fp_mixed_precision(false) { } /** Constructor |