Add FlattenAtrousConv transformation#10996
Conversation
3e5978b to
7becb90
Compare
7becb90 to
8b583d8
Compare
|
I propose to add invocation of this pass into standard optimize function |
|
Interesting, I think we can safely enable this pass in the default pipeline as @elvin-n suggested, @Icemist do you want to do that? You can put the new pass in tvm/src/relay/backend/utils.cc Line 206 in 4c608be |
|
It's pity not to include this pass into standard optimization flows. @Icemist please create another PR with adding it to regular optimization passes. |
Yes, I posted it as a separate commit #11077. I apologize for the long response. |
This transformation flattens atrous convolution, i.e. finds a sequence of space_to_batch_nd->conv2d->batch_to_space_nd operations and convert them into subgraphs with a convolution with the modified "dilation" and recalculated "padding" parameters