Extend definition of grid preserving op into multi-input grid-equivariant op