From 0e7d8b35a2115f2495aa74724a636038255eb7cb Mon Sep 17 00:00:00 2001 From: =?utf8?q?Shubham=20Gupta/System=20SW=20/SRI-Bangalore/Engineer/?= =?utf8?q?=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 24 Oct 2018 09:51:27 +0530 Subject: [PATCH] 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 --- runtimes/pure_arm_compute/src/compilation.cc | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) 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())); -- 2.7.4