Fix return value. (#1010)
authorDenis Maksimenko/AI Tools Lab /SRR/Assistant Engineer/삼성전자 <d.maksimenko@partner.samsung.com>
Tue, 14 Aug 2018 13:43:48 +0000 (16:43 +0300)
committerSergey Vostokov/AI Tools Lab /SRR/Staff Engineer/삼성전자 <s.vostokov@samsung.com>
Tue, 14 Aug 2018 13:43:48 +0000 (16:43 +0300)
createFullyConnected() used to return wrong Node if bias is not present.

Signed-off-by: Denis Maksimenko <d.maksimenko@partner.samsung.com>
contrib/nnc/libs/frontend/caffe/src/caffe_op_creator.cpp

index caf9040..5c2630c 100644 (file)
@@ -310,7 +310,7 @@ std::vector<INode::Ref> OpCreator::createFullyConnected(InputOps &inputs, InputP
   if (!opts.has_bias_term() || opts.bias_term())
     return createOp<ops::BiasAddOp>(fcOutputs, std::move(*params[1]));
   else
-    return outputs;
+    return fcOutputs;
 }
 
 std::vector<INode::Ref> OpCreator::createConcat(InputOps inputs, InputParams params,