From 0194cdce3b324e669612c6a1bb414c8131c603f9 Mon Sep 17 00:00:00 2001 From: Eddy Kim Date: Tue, 15 Aug 2023 15:29:49 +0900 Subject: [PATCH] [GPU] updated to add reorder after parameter when it is connected to quantize (#19182) * updated to add reorder after parameter when it is connected to quantize * removed redundant code --- src/plugins/intel_gpu/src/plugin/ops/parameter.cpp | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/plugins/intel_gpu/src/plugin/ops/parameter.cpp b/src/plugins/intel_gpu/src/plugin/ops/parameter.cpp index c52ff3580dd..74a7e9e8e0a 100644 --- a/src/plugins/intel_gpu/src/plugin/ops/parameter.cpp +++ b/src/plugins/intel_gpu/src/plugin/ops/parameter.cpp @@ -162,6 +162,15 @@ static void CreateParameterOp(Program& p, const std::shared_ptr&)> connected_to_quantize = + [&](const std::shared_ptr &node) -> bool { + for (auto& user : node->get_users()) { + if (ngraph::is_type(user)) + return true; + } + return false; + }; + size_t search_depth = 3; bool is_convert_color_input = recursive_search_convert_color(op, search_depth); bool is_surface_input = has_surface_input(op); @@ -215,7 +224,7 @@ static void CreateParameterOp(Program& p, const std::shared_ptr