Skip to content

Commit ea721f1

Browse files
psunnJerry-Ge
authored andcommitted
[mlir][tosa] Update AVG_POOL2D description to align with TOSAv1.0 Spec
Signed-off-by: Peng Sun <[email protected]> Change-Id: I4ccb6500c81f25d87134bc70a13bdd29034300a2
1 parent d38380d commit ea721f1

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)