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

MLBEDSW-9742 Batched conv2d produces output diff



 - An output diff occurred because decomposition was not done for
convolutions when batch was greater than 1
 - Add check to CanRunOnHardware checking for batch for conv
 - Use SliceShape instead of full shape in DecomposeConv2D and
DecomposeDepthwiseConv2D

Change-Id: Ic932ad329327beea75df8e76b48d737e53aa3cb0
Signed-off-by: Johan Alfvén's avatarJohan Alfven <johan.alfven@arm.com>
parent bf0d7a27
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