The dims for crf and crf_decoding layers are wrong, i.e. height and width are swapped.