Skip to content
Snippets Groups Projects

Use setBackend instead of ConvertToCpu workaround

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

  • Use format HWC for OpenCV tensor and change format when converting to CPU backend.
Edited by Olivier BICHLER

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
Please register or sign in to reply
Loading