File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
lite/backends/nnadapter/nnadapter/src/driver/huawei_ascend_npu Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change 1818#include " driver/huawei_ascend_npu/optimizer/fix_no_inputs_ops.h"
1919#include " driver/huawei_ascend_npu/optimizer/fix_quantized_ops.h"
2020#include " driver/huawei_ascend_npu/optimizer/fix_reduce_ops_scalar_output.h"
21+ #include " optimizer/convert_adaptive_pool2d_into_pool2d.h"
2122#include " optimizer/fuse_conv2d_activation_into_conv2d.h"
2223#include " optimizer/fuse_conv2d_add_into_conv2d.h"
2324#include " optimizer/fuse_conv2d_batch_norm_into_conv2d.h"
@@ -238,6 +239,9 @@ int Program::Build(core::Model* model, core::Cache* cache) {
238239 FuseMatMulDequantAddIntoFullyConnectedDequant (model);
239240 FuseMatMulAddIntoFullyConnected (model, true );
240241 FuseReshapeTransposeReshapeIntoChannelShuffle (model);
242+ #if NNADAPTER_HUAWEI_ASCEND_NPU_CANN_VERSION_GREATER_THAN(6, 0, 0)
243+ ConvertAdaptivePool2dIntoPool2d (model);
244+ #endif
241245 FixQuantizedOps (model);
242246 NNADAPTER_VLOG (5 ) << " Optimized model:" << std::endl << Visualize (model);
243247 // Convert a NNAdapter model to a GE graph
You can’t perform that action at this time.
0 commit comments