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
462 462 * data is copy-transposed.
463 463 */
464 464 void setDataFormat(const DataFormat df, bool copyTrans = true) {
465 if (mImpl && copyTrans && (dataFormat() != df) && df != DataFormat::Default && dataFormat() != DataFormat::Default) {
466 copyTranspose(*this, getDataFormatTranspose(dataFormat(), df));
465 if (!copyTrans || df == dataFormat()) {
466 mDataFormat = df;
467 return;
467 468 }
469
470 if ((df == DataFormat::Default && dataFormat() == DataFormat::Default) ||
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Please register or sign in to reply
    Loading