• H
    support saving model defined parameters without add scale_op (#36119) · 8db6d221
    Haipeng Wang 提交于
    * add scale_op in model save step is not necessary, just fix the prune method to support static graph and inplace op
    
    * fix jit.save, no need to add scale_op to each outputvar anymore.
    fix prune_with_input, now it supports inplace op
    
    * temporarily disable test_trt_dynamic_shape.TRTDynamicShapeOutOfBound2Test
    
    * allow user to export parameters defined in model
    8db6d221
io.py 93.5 KB