Skip to content

Commit

Permalink
Grid sample op (#4373)
Browse files Browse the repository at this point in the history
* pnnx support grid_sample op

* complete the permute and gridsample operator fusion

* spilt calculation into two stages and support permute fusion
  • Loading branch information
Yoh-Z authored Oct 20, 2023
1 parent dc25128 commit 3f437d3
Show file tree
Hide file tree
Showing 17 changed files with 3,374 additions and 201 deletions.
1 change: 1 addition & 0 deletions docs/developer-guide/operators.md
Original file line number Diff line number Diff line change
Expand Up @@ -914,6 +914,7 @@ This function is often used in conjunction with affine_grid() to build Spatial T
| 0 | sample_type | int | 1 | |
| 1 | padding_mode | int | 1 | |
| 2 | align_corner | int | 0 | |
| 3 | permute_fusion| int | 0 | fuse with permute |


Sample type:
Expand Down
Loading

0 comments on commit 3f437d3

Please sign in to comment.