Now forward_dims works for NHWC and NCHW dataformat
Resolves issue #223 (closed)
Conv.hpp
, now Conv_Op::inChannels()
checks if the Weight Producer format is [Cout,H,W,Cin] or [Cout,Cin,H,W];Conv::forwardDims()
accounts for weight data format (NHWC or NCHW)Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent