Add a test of 2D reduction lifting with inner broadcasting
Add tests for the keepdim reduction case
Add tests for 3D lifted reductions, and fix