Skip to content

Commit 3b38992

Browse files
authored
[mlir][tosa] Update AVG_POOL2D description to align with TOSAv1.0 Spec (#129782)
1 parent 2b5ac43 commit 3b38992

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,9 @@ def Tosa_AvgPool2dOp : Tosa_InferShapedTypeOp<"avg_pool2d"> {
7474
let description = [{
7575
This performs an average pooling over the given input tensor. A sliding
7676
window of size given by <kernel size> is passed over the input tensor, with
77-
the mean value being placed in the output tensor.
77+
the mean value being placed in the output tensor. When calculating the
78+
average, only the number of valid input tensor values, but not padding, are
79+
used to calculate the divisor.
7880
}];
7981

8082
let arguments = (ins

0 commit comments

Comments
 (0)