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 Alfven <johan.alfven@arm.com>
Loading
Please register or sign in to comment