// Copyright (C) 2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include <string>
#include <memory>
#include <ngraph/ngraph.hpp>

namespace ngraph {
namespace builder {
namespace subgraph {

class TransformationsAfterSplitFunction {
public:
    static std::shared_ptr<Function> get(const std::string transformationName);

    static std::shared_ptr<Node> getLayerByTransformationName(
        const std::string transformationName,
        const Output<Node> parent);
};

}  // namespace subgraph
}  // namespace builder
}  // namespace ngraph
