SIGN IN SIGN UP

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