Skip to content

Commit

Permalink
[LPT] code style fix #1
Browse files Browse the repository at this point in the history
  • Loading branch information
v-Golubev authored and eshoguli committed Aug 3, 2020
1 parent 3b2ac73 commit 53bd7fe
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
// Copyright (C) 2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include <vector>

#include "layer_transformation.hpp"
#include "ngraph/node.hpp"

namespace ngraph {
namespace pass {
namespace low_precision {

class TRANSFORMATIONS_API SplitTransformation : public LayerTransformation {
public:
SplitTransformation(const Params& params);
void registerMatcherIn(GraphRewrite& pass, TransformationContext& context) const override;
void transform(TransformationContext& context, ngraph::pattern::Matcher& m) const override;
bool isPrecisionPreserved(std::shared_ptr<Node> layer) const noexcept override;
void updateOutputs(
TransformationContext& context,
std::vector<std::shared_ptr<ngraph::Node>> lastNodes,
std::shared_ptr<ngraph::Node> originalNode) const;
protected:
ngraph::Shape getConstSplitShape(
const std::vector<size_t>& constSplitLengths,
const ngraph::Shape& constShape, const size_t axis,
const size_t idx) const;
virtual std::vector<size_t> getConstSplitLengths(
const OutputVector& inputs,
const ngraph::Shape& constShape,
const size_t outputSize) const;
};
} // namespace low_precision
} // namespace pass
} // namespace ngraph
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ class TRANSFORMATIONS_API VariadicSplitTransformation : public SplitTransformati
const OutputVector& inputs,
const ngraph::Shape& constShape,
const size_t outputSize) const override;
};
};
} // namespace low_precision
} // namespace pass
} // namespace ngraph

0 comments on commit 53bd7fe

Please sign in to comment.