commit eaf6e96659957a9120c28035614ba0709767e6c4
author Spenser Bauman <sbauman@mathworks.com> Tue Apr 16 08:14:22 2024 -0400
committer Copybara-Service <copybara-worker@google.com> Tue Apr 16 05:16:11 2024 -0700
tree 2be5bf4b2f25d182cdb26a674061a95cfbf8bda6
parent b0a61843cdd55dc57636b45e129bea94549adbfc

[mlir][tosa] Implement dynamic shape support for tosa.max_pool2d lowering (#87538)

The existing lowering for tosa.max_pool2d only supports dynamic
dimensions when the dynamic dimension is the batch dimension. This
change updates the lowering to support arbitrary dynamic dimensions on
the inputs and outputs of the tosa.max_pool2d operation.

This change also fixes a bug in the implementation of implicit
broadcasting in the tosa-to-linalg pass, which was introducing uses of
constant ops that violated dominance requirements.

GitOrigin-RevId: 1c076b43c294c7d29d99dd50f6853b33a5b99789