From: Shubham Gupta/System SW /SRI-Bangalore/Engineer/삼성전자 Date: Wed, 24 Oct 2018 04:21:27 +0000 (+0530) Subject: Removing TODO comment for reshape (#3309) X-Git-Tag: 0.3~540 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=0e7d8b35a2115f2495aa74724a636038255eb7cb;p=platform%2Fcore%2Fml%2Fnnfw.git Removing TODO comment for reshape (#3309) This patch will remove TODO assert comment of reshape since that TODO is already implemented. Signed-off-by: shubham --- diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc index b0dcf8f..564fb74 100644 --- a/runtimes/pure_arm_compute/src/compilation.cc +++ b/runtimes/pure_arm_compute/src/compilation.cc @@ -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()));