Skip to content

[Feat] ForwardDims Conv for NCHW and NHWC

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