Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
Libnd4j: conv2d op: switch to single weights format #6393
Following on from the discussion in slack: Our libnd4j conv2d op weights format depend on the activations input format (NCHW vs. NHWC).
However, for TF import, we require weights in [kH, kW, inC, outC] for both cases.
Instead, let's change the weights format for conv2d to always be [kH, kW, inC, outC] regardles of the input format.
Note DL4J uses [outC, inC, kH, kW] but we'll handle that at the Java/DL4J level via permute when we use conv2d op.