diff options
| author | Jakub Sujak <jakub.sujak@arm.com> | 2023-03-29 11:16:18 +0100 |
|---|---|---|
| committer | Jakub Sujak <jakub.sujak@arm.com> | 2023-04-04 14:09:21 +0000 |
| commit | 617ed50000532877296fff93973590a8ab67f96d (patch) | |
| tree | d571654a2c31dfb6d5ec1529160b9f3d3348938c /utils | |
| parent | f26ea2f8cc957a1e6faf0361dea805fb2e236061 (diff) | |
| download | ComputeLibrary-617ed50000532877296fff93973590a8ab67f96d.tar.gz | |
Support dynamic weights for Fully Connected layers on GPU
The fully connected function and operator running on GPU have been adapted to support dynamic weights.
Dynamic weights require the reshape and data layout conversion of weight tensors at runtime in the prepare stage of the operator. The implementation for GPU is identical to the CPU implementation.
This patch also deprecates the `are_weights_reshaped` option in Fully Connected.
Resolves: COMPMID-5870
Change-Id: I28f967695879d82cc91a928d95308a4e0e52a597
Signed-off-by: Jakub Sujak <jakub.sujak@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9403
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/TypePrinter.h | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 9b9c7b5b34..d40c3eb3d7 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -2772,7 +2772,6 @@ inline ::std::ostream &operator<<(::std::ostream &os, const FullyConnectedLayerI << "transpose_weights=" << layer_info.transpose_weights << ", " << "are_weights_reshaped=" << layer_info.are_weights_reshaped << ", " << "retain_internal_weights=" << layer_info.retain_internal_weights << ", " - << "constant_weights=" << layer_info.transpose_weights << ", " << "fp_mixed_precision=" << layer_info.fp_mixed_precision << "}"; return os; } |