WrapType Improvements

[GlebKazantaev]

Description

Some pattern based transformations can work with multiple types of operations and to apply the same transformation for two different operation types it was necessary to create two different transformation classes with different patterns but with the same callback or create predicates that check this types. To simplify this workflow WrapType was extended to consume multiple DiscreteTypeInfo instances. Also wrap_type helper functions were updated to consume variadic number of operation types.

Before:

auto is_supported_lstm_cell = [](const std::shared_ptr<Node>& n) {
    return pattern::has_class<ngraph::opset1::LSTMCell>()(n) || pattern::has_class<ngraph::opset4::LSTMCell>()(n);
};
auto any_lstm = std::make_shared<pattern::op::Label>(element::f32, Shape{}, is_supported_lstm_cell);

After:

auto lstm = pattern::wrap_type<opset1::LSTMCell, opset4::LSTMCell>();

Change List

• Updated WrapType/wrap_type facilities to consume multiple types
• GraphRewrite engine was updated to support multiple types in WrapType for efficient execution algorithm
• Simplified transformations with wrap_type facilities
• Removed BatchNormV5Decomposition
• Added tests for WrapType