Implement export-time encoding propagation of grid-equivariant ops