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; }