[Mlir-commits] [mlir] 3b38992 - [mlir][tosa] Update AVG_POOL2D description to align with TOSAv1.0 Spec (#129782)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue Mar 4 17:18:59 PST 2025
Author: Jerry-Ge
Date: 2025-03-05T01:18:56Z
New Revision: 3b38992de1a23f0608a53b2fea9fef836ab0a5c8
URL: https://github.com/llvm/llvm-project/commit/3b38992de1a23f0608a53b2fea9fef836ab0a5c8
DIFF: https://github.com/llvm/llvm-project/commit/3b38992de1a23f0608a53b2fea9fef836ab0a5c8.diff
LOG: [mlir][tosa] Update AVG_POOL2D description to align with TOSAv1.0 Spec (#129782)
Added:
Modified:
mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
Removed:
################################################################################
diff --git a/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td b/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
index 0541bc7f97642..e80805a302e3b 100644
--- a/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
+++ b/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
@@ -74,7 +74,9 @@ def Tosa_AvgPool2dOp : Tosa_InferShapedTypeOp<"avg_pool2d"> {
let description = [{
This performs an average pooling over the given input tensor. A sliding
window of size given by <kernel size> is passed over the input tensor, with
- the mean value being placed in the output tensor.
+ the mean value being placed in the output tensor. When calculating the
+ average, only the number of valid input tensor values, but not padding, are
+ used to calculate the divisor.
}];
let arguments = (ins
More information about the Mlir-commits
mailing list