Since input tensor is modified to NHWC, which is not compatible to cast
kernel in NCHW layout, type_layout_cast_pass will insert a layout
instruction to transform data. Use NHWC cast kernel to avoid cast layout