The “interleaved” artifacts could point towards a layout mismatch (channels-first vs. channels-last) or a view operations, while a permute op would be necessary. Could you check, if the memory layout matches the expected one in the code?