Function conv2d_output_dim

src/framework/modules/conv_modules.cpp:109–111

    "DepthwiseConv2d",
    "nn.conv",
    kConvInputs,
    3,
    kSingleOutput,
    1,
    "Applies a depthwise 2D convolution to channel-first inputs [batch, channels, height, width].",
};