Add comments

microsoft · Nov 29, 2023 · f8dd78c · f8dd78c
1 parent 4ec8086
commit f8dd78c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnxruntime/core/providers/webnn/builders/impl/conv_op_builder.cc b/onnxruntime/core/providers/webnn/builders/impl/conv_op_builder.cc
@@ -252,6 +252,8 @@ Status ConvOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder, const N
         ORT_RETURN_IF_NOT(GetShape(*input_defs[0], input_shape, logger), "Cannot get shape");
         for (size_t i = 0; i < 2; i++) {
           // Get the dimensions of H and W.
+          // For NHWC layout, the dimensions of H and W correspond to index 1 and 2.
+          // For NCHW layout, the dimensions of H and W correspond to index 2 and 3.
           if (model_builder.GetPreferredLayout() == DataLayout::NHWC) {
             total_padding[i] = strides[i] * (narrow<size_t>(input_shape[i + 1]) - 1) +
                                output_padding[i] + ((kernel_shape[i] - 1) * dilations[i] + 1) - output_shape[i];