IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove redundant header (#5363)

Browse Source
This commit is contained in:
Mateusz Bencer 2021-04-23 15:35:46 +02:00 committed by GitHub
parent 0a8c3ef5a9
commit 9a569805c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 105 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
ngraph/frontend/onnx_editor/include/onnx_editor/detail/subgraph_extraction.hpp
View File

@ -1,104 +0,0 @@
// Copyright (C) 2018-2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#pragma once
#include <map>
#include <set>
#include <string>
#include <unordered_map>
#include <vector>
#include "onnx_editor/editor_types.hpp"
namespace ONNX_NAMESPACE
{
class GraphProto;
class NodeProto;
class ValueInfoProto;
} // namespace ONNX_NAMESPACE
namespace ngraph
{
namespace onnx_editor
{
/// \brief Subgraph extraction helper structure
struct SubgraphExtractor
{
SubgraphExtractor(ONNX_NAMESPACE::GraphProto& graph);
/// \brief Adds new inputs to the graph and connects them to the nodes indicated by
/// the provided input edges.
void add_new_inputs(const std::vector<InputEdge>& new_inputs);
/// \brief Adds new outputs to the graph with the same name as the nodes pointed to
/// by the input edges "new_outputs".
void add_new_outputs(const std::vector<OutputEdge>& new_outputs);
/// \brief Extracts the final subgraph by traversing the original model bottom-up
/// starting at each of the provided output edges. The extracted subgraph
/// contains all previously added inputs and potentially a subset of original
/// model's inputs that contribute to the value calculated in the output tensors.
/// In the end the underlying GraphProto is modified and obsolete elements
/// are discarded after this method call has finished.
///
/// \param subgraph_outputs A list of expected outputs of the extracted subgraph.
void extract_subgraph(std::vector<OutputEdge> subgraph_outputs);
/// \brief Represents a subgraph of an ONNX model by holding a subset of nodes, inputs,
/// outputs and initializers of the original graph. Objects of this struct can be
/// merged into other instances using the += operator to build a subgraph from
/// smaller clusters.
struct SubgraphComponents
{
SubgraphComponents() = default;
SubgraphComponents(const SubgraphComponents&) = delete;
SubgraphComponents(SubgraphComponents&&) = default;
SubgraphComponents& operator=(const SubgraphComponents&) = delete;
SubgraphComponents& operator=(SubgraphComponents&&) = default;
std::set<int> nodes;
std::set<std::string> inputs;
std::set<std::string> initializers;
std::set<std::string> outputs;
SubgraphComponents& operator+=(SubgraphComponents&& other)
{
nodes.insert(other.nodes.begin(), other.nodes.end());
inputs.insert(other.inputs.begin(), other.inputs.end());
initializers.insert(other.initializers.begin(), other.initializers.end());
outputs.insert(other.outputs.begin(), other.outputs.end());
return *this;
}
};
private:
ONNX_NAMESPACE::GraphProto& m_onnx_graph;
// Graph traversal helper: node index -> node inputs (one-to-many)
std::unordered_multimap<int, std::string> m_node_inputs;
// Number of consumers of all tensors in the graph
std::map<std::string, int> m_tensor_consumers;
/// \brief Replaces the old input edge with a new one in the helper struct.
/// This is used by the output contributors discovery.
void replace_input_edge(const InputEdge& old_edge, const InputEdge& new_edge);
/// \brief Returns a list of edges of each outputs of the graph "m_onnx_graph"
std::vector<OutputEdge> all_output_edges() const;
/// \brief Traverses the graph bottom-up and collects all nodes, inputs and initializers
/// that contribute to an output designated by the provided output edge.
/// A sum of such SubgraphComponents objects forms a target extracted subgraph.
SubgraphComponents
discover_output_contributors(const OutputEdge& output_edge,
const SubgraphComponents& already_collected) const;
/// \brief Modifies the underlying GraphProto object and discards all obsolete elements.
///
/// \param subgraph An object describing the subgraph to be extracted (elems to be kept)
void extract_subgraph_from_onnx_model(const SubgraphComponents& subgraph);
};
} // namespace onnx_editor
} // namespace ngraph

2
ngraph/frontend/onnx_editor/src/detail/subgraph_extraction.cpp
View File

@ -7,7 +7,7 @@
#include <stack>
#include "ngraph/check.hpp"
#include "onnx_editor/detail/subgraph_extraction.hpp"
#include "subgraph_extraction.hpp"
using namespace ngraph::onnx_editor;