Fix build issue #2668 (#2669)
authorСергей Баранников/AI Tools Lab /SRR/Engineer/삼성전자 <s.barannikov@samsung.com>
Thu, 13 Dec 2018 12:02:58 +0000 (15:02 +0300)
committerEfimov Alexander/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Thu, 13 Dec 2018 12:02:58 +0000 (15:02 +0300)
- Build failure issued by merging of #2646: convertBlob return value changed
- Small code style changes

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>
contrib/nnc/include/core/modelIR/operations/ConstantOp.h
contrib/nnc/passes/caffe_frontend/caffe_op_creator.cpp

index 92d22f4..46d4125 100644 (file)
@@ -25,15 +25,15 @@ namespace ops {
 
 class ConstantOp : public Operation {
 public:
-  ConstantOp(const std::shared_ptr<mir::TensorVariant>& value) :
-                                            Operation(Type::constant, {}), _value(value) {
+  ConstantOp(const std::shared_ptr<mir::TensorVariant>& value)
+      : Operation(Type::constant, {}), _value(value) {
     setOutputShape(0, _value->getShape());
   }
 
-    const std::shared_ptr<mir::TensorVariant>& getValue() const { return _value; }
+  const std::shared_ptr<mir::TensorVariant>& getValue() const { return _value; }
 
 private:
-    const std::shared_ptr<mir::TensorVariant>& _value;
+  const std::shared_ptr<mir::TensorVariant>& _value;
 };
 
 } // namespace ops
index 6657ec9..e149d53 100644 (file)
@@ -547,7 +547,7 @@ std::vector<mir::IODescriptor>
 CaffeOpCreator::convertEmbed(const caffe::LayerParameter& layer,
                              const std::vector<mir::IODescriptor>& inputs) {
   const auto& params = layer.embed_param();
-  auto data = createOp<ops::ConstantOp>(layer.name() + ".weights", *convertBlob(layer.blobs(0)));
+  auto data = createOp<ops::ConstantOp>(layer.name() + ".weights", convertBlob(layer.blobs(0)));
   // FIXME Indices in Caffe have floating type, while in ModelIR they are integral.
   auto result = createOp<ops::GatherOp>(layer.name(), data->getOutput(0), inputs[0], 0);