Given the different backends dimension order, we should be able to prepare the tensors on HxWxC formatting or CxHxW formatting.