Skip to content

Commit

Permalink
fix:pnnx-softmax (#4333)
Browse files Browse the repository at this point in the history
  • Loading branch information
EdVince committed Nov 5, 2022
1 parent 92da26b commit abb2843
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/pnnx/src/pass_level2/F_softmax.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class F_softmax : public GraphRewriterPass
5 4
pnnx.Input input_0 0 1 input
pnnx.Input input_1 0 1 dim
prim::Constant op_0 0 1 dtype value=None
prim::Constant op_0 0 1 dtype value=*
aten::softmax op_1 3 1 input dim dtype out
pnnx.Output output 1 0 out
)PNNXIR";
Expand Down

0 comments on commit abb2843

Please sign in to comment.