diff options
author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2023-06-27 14:25:58 +0100 |
---|---|---|
committer | Mohmun02 <MohammedSuhail.Munshi@arm.com> | 2023-07-11 08:53:19 +0000 |
commit | 8e2dedea8550b1c18c3bbeead8c972f661dcfac8 (patch) | |
tree | 61cd0326b9690e343d62a5c72d935fcd68017eb9 /src/gpu/cl/operators/ClFullyConnected.h | |
parent | 5ff480265a110ea1f2ce24491e082f52348b0f92 (diff) | |
download | ComputeLibrary-8e2dedea8550b1c18c3bbeead8c972f661dcfac8.tar.gz |
Add Bias to MatMul Kernels and add support for use in Fully Connected Layer
Resolves: [COMPMID-6316]
Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com>
Change-Id: I08e6bac9e6b46b76978da0dc6a48ccfe3dde5086
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9833
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/gpu/cl/operators/ClFullyConnected.h')
-rw-r--r-- | src/gpu/cl/operators/ClFullyConnected.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/src/gpu/cl/operators/ClFullyConnected.h b/src/gpu/cl/operators/ClFullyConnected.h index 9a5ba40510..5a71bd24c7 100644 --- a/src/gpu/cl/operators/ClFullyConnected.h +++ b/src/gpu/cl/operators/ClFullyConnected.h @@ -132,17 +132,18 @@ private: TensorInfo _flattened_src{}; TensorInfo _converted_weights{}; TensorInfo _reshaped_weights{}; - TensorInfo _lhs_to_use{}; + TensorInfo _lhs_to_use{}; TensorInfo _weights_to_use{}; int _weights_to_use_idx{ ACL_SRC_1 }; - bool _are_weights_converted{ true }; - bool _are_weights_reshaped{ true }; + bool _run_convert_weights{ false }; + bool _transpose_weights{ false }; + bool _dynamic_gemm{ false }; + bool _use_matmul{ false }; + bool _is_fc_after_conv{ true }; bool _is_quantized{ false }; bool _is_prepared{ false }; - bool _dynamic_weights{ false }; - bool _use_matmul{ false }; #ifdef ARM_COMPUTE_ASSERTS_ENABLED int _asrt_run_count {}; |