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
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);