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