Removing TODO comment for reshape (#3309)
authorShubham Gupta/System SW /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Wed, 24 Oct 2018 04:21:27 +0000 (09:51 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Wed, 24 Oct 2018 04:21:27 +0000 (13:21 +0900)
This patch will remove TODO assert comment of reshape since that TODO is already implemented.

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

index b0dcf8f..564fb74 100644 (file)
@@ -2254,17 +2254,13 @@ void Planner::visit(const ::internal::tflite::op::Reshape::Node &node)
   const ::internal::tflite::operand::Index output_index{node.param().output_index};
   const ::internal::tflite::operand::Index input_index{node.param().input_index};
 
-  // TODO Re-enable this assert
-  // assert((ifm_shape.C * ifm_shape.H * ifm_shape.W) == out_size);
-
-  // TODO Should move to the place where the operand is handled, if it is possible.
-
   auto input_shape = asTensorShape(_ctx.at(input_index).shape());
   auto output_shape = asTensorShape(_ctx.at(output_index).shape());
 
   assert(input_shape[0] * input_shape[1] * input_shape[2] * input_shape[3] ==
          output_shape[0] * output_shape[1] * output_shape[2] * output_shape[3]);
 
+  // TODO Should move to the place where the operand is handled, if it is possible.
   _builder.addShapeConstr(output_index, asTensorInfo(output_shape, _ctx.at(output_index).type(),
                                                      _ctx.at(output_index).scale(),
                                                      _ctx.at(output_index).zeroPoint()));