Removing TODO for pool op (#3370)
authorShubham Gupta/System SW /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Wed, 31 Oct 2018 01:24:50 +0000 (06:54 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Wed, 31 Oct 2018 01:24:50 +0000 (10:24 +0900)
This patch will remove TODO statement for max_pool and avg_pool

Signed-off-by: shubham <shub98.gupta@samsung.com>
runtimes/pure_arm_compute/src/compilation.cc

index 14e9452..5727846 100644 (file)
@@ -1700,7 +1700,6 @@ void Planner::visit(const ::internal::tflite::op::MaxPool2D::Explicit::Node &nod
 
   const ::internal::tflite::operand::Index activation_index{node.param().activation_index};
 
-  // TODO 4D tensor (dim(0) !=1 )
   const auto ofm_shape = _ctx.at(ofm_index).shape().asFeature();
   const auto ifm_shape = _ctx.at(ifm_index).shape().asFeature();
 
@@ -1934,7 +1933,6 @@ void Planner::visit(const ::internal::tflite::op::AvgPool2D::Explicit::Node &nod
 
   const ::internal::tflite::operand::Index activation_index{node.param().activation_index};
 
-  // TODO 4D tensor (dim(0) != 1)
   const auto ofm_shape = _ctx.at(ofm_index).shape().asFeature();
   const auto ifm_shape = _ctx.at(ifm_index).shape().asFeature();