|
|
|
@@ -63,6 +63,29 @@ pnnx.Output output 1 0 out |
|
|
|
|
|
|
|
REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(F_upsample_nearest_1, 10) |
|
|
|
|
|
|
|
class F_upsample_nearest_1_1 : public GraphRewriterPass |
|
|
|
{ |
|
|
|
public: |
|
|
|
const char* match_pattern_graph() const |
|
|
|
{ |
|
|
|
return R"PNNXIR(7767517 |
|
|
|
5 4 |
|
|
|
pnnx.Input input_0 0 1 input |
|
|
|
pnnx.Input input_1 0 1 size |
|
|
|
prim::Constant op_0 0 1 scale_factor value=None |
|
|
|
aten::upsample_nearest2d op_1 3 1 input size scale_factor out |
|
|
|
pnnx.Output output 1 0 out |
|
|
|
)PNNXIR"; |
|
|
|
} |
|
|
|
|
|
|
|
const char* type_str() const |
|
|
|
{ |
|
|
|
return "F.upsample_nearest"; |
|
|
|
} |
|
|
|
}; |
|
|
|
|
|
|
|
REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(F_upsample_nearest_1_1, 10) |
|
|
|
|
|
|
|
class F_upsample_nearest_2 : public GraphRewriterPass |
|
|
|
{ |
|
|
|
public: |
|
|
|
|