ResNet-50 with different image/kernel layouts

I’d like to generate a ResNet-50 with image layout NHWC and kernel layout HWIO. It seems like the current testing infrastructure (relay/testing/resnet.py) doesn’t support this. I could modify it to support this (by just passing layout/kernel_layout args through), but I’m wondering if I’ll potentially run into other problems down the road (i.e. Relay complaining about layouts not lining up, internally). Is there an easier way to get what I need?