diff options
author | Milos Puzovic <Milos.Puzovic@arm.com> | 2022-10-28 16:49:15 +0100 |
---|---|---|
committer | Milos Puzovic <milos.puzovic@arm.com> | 2022-11-01 15:04:10 +0000 |
commit | a7077e9b8cc2d93a84249ee665776d09963e08a0 (patch) | |
tree | 2f810da77d32380e65f6561f80f2dcd2404beb86 /src/cpu/operators/CpuDepthwiseConv2d.h | |
parent | 199982fc01bcce10120cd6df03d11829a215c85c (diff) | |
download | ComputeLibrary-a7077e9b8cc2d93a84249ee665776d09963e08a0.tar.gz |
Updateable weights in depthwise convolution
Check whether weights are defined as constant, if they are not constant then do not pack them if they are already packed so that they can be updated.
Signed-off-by: Milos Puzovic <Milos.Puzovic@arm.com>
Change-Id: I73447e31e3660b05f8f40e04ea4ea2003eb9b802
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8539
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/operators/CpuDepthwiseConv2d.h')
-rw-r--r-- | src/cpu/operators/CpuDepthwiseConv2d.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/cpu/operators/CpuDepthwiseConv2d.h b/src/cpu/operators/CpuDepthwiseConv2d.h index 15e52ef515..3d8719ee44 100644 --- a/src/cpu/operators/CpuDepthwiseConv2d.h +++ b/src/cpu/operators/CpuDepthwiseConv2d.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. + * Copyright (c) 2021-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -143,6 +143,7 @@ private: bool _permute{ false }; bool _is_activationlayer_enabled{ false }; bool _is_prepared{ false }; + bool _are_weights_const{ true }; }; /** Basic function to execute a generic depthwise convolution. This function calls the following kernel: |