Arm backend: Add fuse batchnorm pass (#8028)
* Support testing passes needing ExportedProgram ExportPasses that need to be initated with an ExportedProgram can currently not be tested in a convenient way. This patch subclasses RunPasses, adding a parameter, `passes_with_exported_program`, that can be used just like `pass_list` but are initiated with an exported program before they are run. The functionality is tested in new tests for the CastInt64Pass and InsertTableOpsPass * Add pass for fusing batchnorm into conv The pass differs from existing fuse passes since they use the get_attr node which is not supported by ArmBackend. Instead, we update the existing parameters. Also adds tests. Signed-off-by: Erik Lundell <erik.lundell@arm.com>
E
Erik Lundell committed
5f849ca02f61a42a22e53f25d3385bb7282fdec8
Parent: a373925
Committed by GitHub <noreply@github.com>
on 1/31/2025, 9:59:47 AM