I’m not sure, if reshaping the batch size into the channels is the right approach here.
Currently you are using an own filter for each channel in each batch (128*8).
This would set your batch size to a fixed value, which seems a bit unusual.
Could you explain your use case a bit?