• S
    fix tensorrt output varible reshape (#31733) · 9d04ef73
    Shang Zhizhou 提交于
    * fix tensorrt output varible reshape
    
    * move padding shape x 1 x 1 in ernie to qkv and fc
    
    * update layer name
    
    * fix softmax when input is dynamic, fc not padding any more
    
    * fix varlen
    
    * move fc x_dim assert to op_teller
    9d04ef73
softmax_op.cc 2.8 KB