MNN_NPU IR op modify

alibaba · Feb 20, 2024 · 241289e · 241289e
1 parent 24841e7
commit 241289e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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;
  }