• L
    Cherry pick bug fix for Ops: reshape,concat, split and squeeze (#20929) · 33d7aae1
    liym27 提交于
    * [cherry-pick]fix bug in reshape: (#20781)
    
    consider the situation that shape of input can contain more than one -1.
    
    * [cherry-pick]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.
    * redefine function GetDataFromTensor and set unknown output shape to - 1.
    * add check: Attr(sections) match Input(X).
    * support Tensor for attr(sections) and attr(sections) can contain -1.
    * modify error message and fix bug for concat and call Resize only when necessary.
    test=release/1.6
    
    * [cherry-pick]improve unsqueeze op to support int, Tensor for argument axes (#20824)
    
    * improve unsqueeze op to support int, Tensor and Tensor list for argument axes.
    * call Resize only when necessary. test=release/1.6
    
    * [cherry-pick]Compatible int32 and int64 for attr in concat/split/unsqueeze. test=release/1.6 (#20912)
    33d7aae1
nn.py 702.1 KB