Skip to content

Draft: Use setBackend instead of ConvertToCpu workaround

Thibault Allenet requested to merge setBackend into dev

Change the convertion to backend 'cpu' from a custom function converToCpu in the database to the generic setBackend().

Content :

  • Remove the call to convertToCpu in database, its definition & tests

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].

Merge request reports