Channel Group Normalization - Coding problem | OneNoughtOne