Questions about conv2d in caffe2 and fbgemm

Hi, in caffe2/quantization/server/conv_dnnlowp_op.cc, i find it says “only 1-3d convolution supported by NHWC”, so only 3d convolution can be implemented and accelerated by fbgemm? how to apply fbgemm to 2d convolution?

This means 1d, 2d and 3d are supported for NHWC storage. 2d convolutions (for low precision) are accelerated by fbgemm when the engine used is DNNLOWP.