NHWC inputs/outputs

Please refer,