test/verify/test_unsqueeze_conv_relu.cpp · 4ea391163f2318bb76b833aa0f40a306301e65db · gaoqiong / MIGraphX

Fix MLIR input fusion non-std shapes from squeeze, flatten and unsqueeze (#2313) · 1a1c1b42

Manupa Karunaratne authored Oct 12, 2023

Currently, we see MLIR partition candidates recieving non-standard shape due to not fusing in squeeze, flatten and unsqueeze ops. These ops could be canonicalized to reshape without introducing additional ops as long as MLIR backend is concerned.

1a1c1b42

test_unsqueeze_conv_relu.cpp 1.98 KB

Replace test_unsqueeze_conv_relu.cpp