[Fix] support variable steps' slice operation of antares ir

microsoft / nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.

MIT License

948 stars 158 forks source link

[Fix] support variable steps' slice operation of antares ir #449

Closed LeiWang1999 closed 2 years ago

LeiWang1999 commented 2 years ago

swin-transformer-tiny has a lot of slice nodes which's step attribute is 2, but currently our code only support slice operation with step equal to 1, this pull request adds support of variable steps' slice operation antares ir generation.