Imported Upstream version 1.7.0

[platform/core/ml/nnfw.git] / compiler / locomotiv / src / Node / AvgPool2D.cpp

diff --git a/compiler/locomotiv/src/Node/AvgPool2D.cpp b/compiler/locomotiv/src/Node/AvgPool2D.cpp
index ad603ba..5fdf1e7 100644 (file)
--- a/compiler/locomotiv/src/Node/AvgPool2D.cpp
+++ b/compiler/locomotiv/src/Node/AvgPool2D.cpp
@@ -129,7 +129,8 @@ nncc::core::ADT::tensor::Buffer<T> avgPool2D(const loco::AvgPool2D *avgpool2d,
             }
           }
 
-          assert(filter_ele_count > 0);
+          if (filter_ele_count <= 0)
+            throw std::runtime_error("The number of filter element must be greater than zero.");
           output_buf.at(Index({batch, out_y, out_x, channel})) = total / filter_ele_count;
         }
       }
@@ -141,10 +142,12 @@ nncc::core::ADT::tensor::Buffer<T> avgPool2D(const loco::AvgPool2D *avgpool2d,
 
 } // namespace
 
-namespace locomotiv
+namespace
 {
 
-void NodeExecution::execute(loco::AvgPool2D *avgpool2d)
+using namespace locomotiv;
+
+void exectute_node(loco::AvgPool2D *avgpool2d)
 {
   auto ifm_data = annot_data(avgpool2d->ifm());
 
@@ -176,4 +179,11 @@ void NodeExecution::execute(loco::AvgPool2D *avgpool2d)
   annot_domain(avgpool2d, loco::Domain::Feature);
 }
 
+} // namespace
+
+namespace locomotiv
+{
+
+void NodeExecution::execute(loco::AvgPool2D *avgpool2d) { exectute_node(avgpool2d); }
+
 } // namespace locomotiv