• Z
    fix the bug of conv_transpose cudnn kernel, test=develop (#20958) · f4f85831
    Zhang Ting 提交于
    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.
    f4f85831
conv_transpose_cudnn_op.cu 24.4 KB