-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
public:
explicit FlattenShapeProp(const std::string &type) : BuiltInShapeInferImpl(type) {}
- void inferShapesImpl(const std::vector<SizeVector> &inShapes,
- const std::map<std::string, std::string> ¶ms,
- const std::map<std::string, Blob::Ptr> &blobs,
- std::vector<SizeVector> &outShapes) override {
+ void inferShapesImpl(const std::vector<Blob::CPtr>& inBlobs,
+ const std::map<std::string, std::string>& params,
+ const std::map<std::string, Blob::Ptr>& blobs,
+ std::vector<SizeVector>& outShapes) override {
LayerParams lp{};
ReshapeLayer reshapeLayer(lp);
reshapeLayer.params = params;
reshapeLayer.type = _type;
- validate(&reshapeLayer, inShapes, params, blobs);
+ validate(&reshapeLayer, inBlobs, params, blobs);
auto inputShape = inShapes[0];
size_t inputShapeTotal = std::accumulate(inputShape.begin(), inputShape.end(), 1lu, std::multiplies<size_t>());