[nnc] Fixed weights tensor shape in ACL backend (#2958)
authorIvan Vagin/AI Tools Lab /SRR/Engineer/삼성전자 <ivan.vagin@samsung.com>
Tue, 29 Jan 2019 15:07:48 +0000 (18:07 +0300)
committerEfimov Alexander/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Tue, 29 Jan 2019 15:07:48 +0000 (18:07 +0300)
Fixed name of weights tensor shape variable in ACL backend to fix build

Signed-off-by: Ivan Vagin <ivan.vagin@samsung.com>
contrib/nnc/passes/acl_soft_backend/AclCppOpGenerator.cpp

index 12501cd..7f5ec8c 100644 (file)
@@ -338,7 +338,7 @@ void AclCppOpGenerator::visit(ops::FullyConnectedOp& op) {
     throw AclCppException("Unsupported operation type");
 
   const TensorVariant ir_weights_tensor = transposeTensor<1, 0>(ir_weights_op->getValue());
-  const Shape& ir_weights_shape = ir_weights.getShape();
+  const Shape& ir_weights_shape = ir_weights_tensor.getShape();
 
   // Get the input node tensor id in the DOM.
   auto in = AF::id(tensorName(ir_input));