Skip to content
Commit 3696cf32 authored by Johan Alfvén's avatar Johan Alfvén
Browse files

MLBEDSW-9764: Optimize 1x1 Conv2D as FullyConnected



 - Batched Conv2D with kernel 1x1 can be optimized in the same way
as FullyConnected
 - Add condition to RewriteFullyConnected to detect the Conv2D case
 - Moved condition to scheduler when to buffer for FullyConnect
 - Extend CanSubdivide with optype FullyConnected
 - Set AxisOrder in TOSA reader for Conv2D

Change-Id: I8ecc9c5bb8700a00e18d54cb8cc209ab70d01f45
Signed-off-by: Johan Alfvén's avatarJohan Alfven <johan.alfven@arm.com>
parent 4541858f
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment