The 2nd element of the shape of inputs and kernels in the ConvOp should be called the number of* channels* rather than the img_height or stackheight or whatever.