Skip to content
Snippets Groups Projects

[Feat] ForwardDims Conv for NCHW and NHWC

Merged Wissam Boussella requested to merge wboussella/aidge_core:nhwc_for_conv into dev

Context

Now forward_dims works for NHWC and NCHW dataformat

Resolves issue #223 (closed)

Modified files

  • feat: Conv.hpp, now Conv_Op::inChannels() checks if the Weight Producer format is [Cout,H,W,Cin] or [Cout,Cin,H,W];
  • upd: Conv::forwardDims() accounts for weight data format (NHWC or NCHW)
Edited by Wissam Boussella

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