tree 0c5d63bbbcc285232959a5a4134f282a980ab4bf
parent 511771fbe0a74e6d9dfd37ba9b4926a8315ec7aa
author Georgios Pinitas <georgios.pinitas@arm.com> 1629476805 +0100
committer SiCong Li <sicong.li@arm.com> 1629795623 +0000

Re-use auxiliary memory withing CpuWinogradConv2d operators

Input/Output transformation operations are independent and done in
different time-steps of the algorithm, this memory can be re-used
between this transformation stages.

Moreover, reduce the allocation when extracting workspace sizes for
Winograd trasformations. There is a mix return of sizes in bytes and
elements, thus ensure the correct is in place. storage_size() member
functions return elements while working_space() function bytes.

Resolves: COMPMID-4781

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I705445ba7ca818cead48369db3cacd49684c7192
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6145
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>