Skip to content

Commit

Permalink
w
Browse files Browse the repository at this point in the history
  • Loading branch information
nihui committed Feb 10, 2025
1 parent 22a5b7b commit 28302af
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions tools/pnnx/src/pass_level2/torch_index_select.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -45,13 +45,19 @@ class torch_index_select_tnn : public GraphRewriterPass
public:
const char* match_pattern_graph() const
{
// clang-format off
// *INDENT-OFF*

return R"PNNXIR(7767517
4 3
pnnx.Input input 0 1 input
pnnx.Attribute op_0 0 1 index @data=(?)i32
tnn.Gather op_1 2 1 input index out arg0=%dim arg1=0 arg2=1
pnnx.Output output 1 0 out
)PNNXIR";

// *INDENT-ON*
// clang-format on
}

const char* replace_pattern_graph() const
Expand Down

0 comments on commit 28302af

Please sign in to comment.