Use setBackend instead of ConvertToCpu workaround
Compare changes
Files
4@@ -33,27 +33,6 @@ namespace Aidge {
\ No newline at end of file
Change the convertion to backend 'cpu' from a custom function converToCpu
in the database to the generic setBackend()
.
Content :
However it raises a problem on the layout of the data between different backends.
In the current code the resulting tensor with backend 'cpu' have layout [N,H,W,C] instead of [N,C,H,W].
Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent