diff options
author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-03-13 16:20:04 +0000 |
---|---|---|
committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-03-21 10:33:53 +0000 |
commit | a3e57c20a0b7a174f0c357676a4da40a248d04db (patch) | |
tree | d92b2316a00db6ce07dd2af476791281fcc98de6 /src/cpu/operators/CpuFullyConnected.h | |
parent | 8918b23073851417e8be6e5e53c6380dbdedf201 (diff) | |
download | ComputeLibrary-a3e57c20a0b7a174f0c357676a4da40a248d04db.tar.gz |
Add dynamic weights for CPU fully connected layer
Resolves: COMPMID-5917
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I073067b490f2a1b96b81a037ea431c9a2e5c7503
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9322
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/operators/CpuFullyConnected.h')
-rw-r--r-- | src/cpu/operators/CpuFullyConnected.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/cpu/operators/CpuFullyConnected.h b/src/cpu/operators/CpuFullyConnected.h index 9cd67f2ca6..a5a464f67a 100644 --- a/src/cpu/operators/CpuFullyConnected.h +++ b/src/cpu/operators/CpuFullyConnected.h @@ -155,6 +155,12 @@ private: bool _enable_fast_math; bool _fixed_format; arm_compute::WeightFormat _weight_format; + bool _dynamic_weights; + +#ifdef ARM_COMPUTE_ASSERTS_ENABLED + int _asrt_run_count{}; + int _asrt_prepare_count{}; +#endif // ARM_COMPUTE_ASSERTS_ENABLED }; } // namespace cpu } // namespace arm_compute |