fix the bug of conv_transpose cudnn kernel, test=develop (#20958)
fix the bug of conv_transpose cudnn kernel: before version 1.6, the data_format is AnyLayout in inference model. When use version 1.6 and load the model which is saved by previous version, the error occurs. This is because the cudnn kernel in version 1.6 is not compitable with Anylayout setting.custom_op_abi
parent
7695b713e1
commit
f4f85831d3
Loading…
Reference in new issue