[GPU] Fuse MVN before ConvertPrecision (#18845)

This commit is contained in:
Pavel Esir 2023-07-28 20:05:05 +02:00 committed by GitHub
parent 4c1fc5ff47
commit 5113900350
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -70,6 +70,7 @@
#include <transformations/op_conversions/lstm_cell_decomposition.hpp>
#include <transformations/op_conversions/rnn_cell_decomposition.hpp>
#include <transformations/op_conversions/mvn6_decomposition.hpp>
#include <transformations/common_optimizations/mvn_fusion.hpp>
#include <transformations/op_conversions/normalize_l2_decomposition.hpp>
#include <transformations/op_conversions/bidirectional_sequences_decomposition.hpp>
#include <transformations/op_conversions/convert_previous_nms_to_nms_9.hpp>
@ -193,9 +194,10 @@ void TransformationsPipeline::apply(std::shared_ptr<ov::Model> func) {
type_to_fuse_map empty_fuse_map = {};
manager.register_pass<ov::pass::Validate>();
// fuse softmax patterns so that they will not be marked as precision sensitive in ConvertPrecision
// fuse softmax, MVN patterns, so that they will not be marked as precision sensitive in ConvertPrecision
manager.register_pass<ov::pass::SoftmaxFusion>();
// decompose MVNs that sre not supported in GPU, so the they will be marked as precision sensitive in ConvertPrecision
manager.register_pass<ov::pass::MVNFusion>();
// decompose MVNs that sre not supported in GPU, so that they will be marked as precision sensitive in ConvertPrecision
manager.register_pass<ov::pass::MVN6Decomposition>();
manager.register_pass<ov::pass::BroadcastTransition>();