• L
    support Tensor for split and concat, support -1 in num_or_sections, add check... · 6802539a
    liym27 提交于
    support Tensor for split and concat, support -1 in num_or_sections, add check num_or_sections (#20780)
    
    * improve split and concat op:
    1. support Tensor for argument 'dim' in split op.
    2. support Tensor for argument 'axis' in concat op.
    test=develop
    
    * redefine function GetDataFromTensor and set unknown output shape to - 1.
    test=develop
    
    * add check: Attr(sections) match Input(X). test=develop
    
    * support Tensor for attr(sections) and attr(sections) can contain -1.
    add check for attr(sections).
    test=develop
    
    * modify error message for concat and call Resize only when necessary. test=develop
    6802539a
concat_op.h 6.5 KB