From 41671dbc24a64742cdd3c0f54b268a2fd2e5a12a Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9E=A5=EC=A7=80=EC=84=AD/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Mon, 11 Jun 2018 16:24:09 +0900 Subject: [PATCH] Add quantization information to TensorInfo of CAST (#1646) This commit adds quantization information to TensorInfo of CAST. Signed-off-by: jiseob.jang --- runtimes/pure_arm_compute/src/compilation.cc | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc index cb408d5..76efc4f 100644 --- a/runtimes/pure_arm_compute/src/compilation.cc +++ b/runtimes/pure_arm_compute/src/compilation.cc @@ -1481,24 +1481,36 @@ void Planner::visit(const ::internal::tflite::op::Cast::Node &node) { case 0: // scalar { - _builder.addShapeConstr(output_index, asTensorInfo(1, _ctx.at(output_index).type())); - _builder.addShapeConstr(input_index, asTensorInfo(1, _ctx.at(input_index).type())); + _builder.addShapeConstr(output_index, asTensorInfo(1, _ctx.at(output_index).type(), + _ctx.at(output_index).scale(), + _ctx.at(output_index).zeroPoint())); + _builder.addShapeConstr(input_index, asTensorInfo(1, _ctx.at(input_index).type(), + _ctx.at(input_index).scale(), + _ctx.at(input_index).zeroPoint())); break; } case 1: // vector { _builder.addShapeConstr(output_index, - asTensorInfo(input_shape.asVector(), _ctx.at(output_index).type())); + asTensorInfo(input_shape.asVector(), _ctx.at(output_index).type(), + _ctx.at(output_index).scale(), + _ctx.at(output_index).zeroPoint())); _builder.addShapeConstr(input_index, - asTensorInfo(output_shape.asVector(), _ctx.at(input_index).type())); + asTensorInfo(output_shape.asVector(), _ctx.at(input_index).type(), + _ctx.at(input_index).scale(), + _ctx.at(input_index).zeroPoint())); break; } case 4: // feature { _builder.addShapeConstr(output_index, - asTensorInfo(input_shape.asFeature(), _ctx.at(output_index).type())); + asTensorInfo(input_shape.asFeature(), _ctx.at(output_index).type(), + _ctx.at(output_index).scale(), + _ctx.at(output_index).zeroPoint())); _builder.addShapeConstr(input_index, - asTensorInfo(output_shape.asFeature(), _ctx.at(input_index).type())); + asTensorInfo(output_shape.asFeature(), _ctx.at(input_index).type(), + _ctx.at(input_index).scale(), + _ctx.at(input_index).zeroPoint())); break; } default: -- 2.7.4