Browse Source

f

pull/6233/head
nihuini 11 months ago
parent
commit
fc2951ecf6
No known key found for this signature in database GPG Key ID: 98FD8F4EBC3E5DB8
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      tools/pnnx/src/pass_level2/torch_flip.cpp

+ 1
- 2
tools/pnnx/src/pass_level2/torch_flip.cpp View File

@@ -49,12 +49,11 @@ pnnx.Output output 1 0 out
{
if (captured_params.at("axes").type == 2)
{
int axis = captured_params.at("axes").i;
int start = captured_params.at("starts").i;
int end = captured_params.at("ends").i;
int step = captured_params.at("steps").i;

if (axis == 0 && start == -1 && end == INT_MIN + 1 && step == -1)
if (start == -1 && end == INT_MIN + 1 && step == -1)
return true;
}
else // if (captured_params.at("axes").type == 5)


Loading…
Cancel
Save