Tensorflow Group Convolution