[Feat] ForwardDims Conv for NCHW and NHWC
Context
Now forward_dims works for NHWC and NCHW dataformat
Resolves issue #223 (closed)
Modified files
- feat:
Conv.hpp
, nowConv_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
Activity
Filter activity
Please register or sign in to reply