Both input formats are supported, but NHWC gives the best performance. Habana devices effectively support input tensors with an NHWC layout, meaning that only tensors whose channel component is the fastest changing and in dimension 0.
Both input formats are supported, but NHWC gives the best performance. Habana devices effectively support input tensors with an NHWC layout, meaning that only tensors whose channel component is the fastest changing and in dimension 0.