From 241289e43d788f3fae992b96f4e90b64dee82b6a Mon Sep 17 00:00:00 2001
From: 946166920 <412384199@qq.com>
Date: Tue, 20 Feb 2024 11:19:54 +0800
Subject: [PATCH] MNN_NPU IR op modify

---
 source/shape/ShapePool3D.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/source/shape/ShapePool3D.cpp b/source/shape/ShapePool3D.cpp
index a346b790d..dc55f7fd5 100644
--- a/source/shape/ShapePool3D.cpp
+++ b/source/shape/ShapePool3D.cpp
@@ -77,6 +77,9 @@ class Pool3DSizeComputer : public SizeComputer {
         auto size  = (float)outputs[0]->elementSize() / 1024.0f / 1024.0f;
         auto layer = op->main_as_Pool3D();
         float flopsPerElement = 1;
+        if (layer->kernels() == nullptr) {
+            return size * flopsPerElement;
+        }
         for (auto kernel: *layer->kernels()) {
             flopsPerElement *= kernel;
         }