[GPU] fix Transpose issue for ConvertColor with FakeQuantize. (#12645) (

#12761) Co-authored-by: Tang Wei <[email protected]> Co-authored-by: Kurt Chen <[email protected]>
openvinotoolkit · Aug 26, 2022 · 41a404f · 41a404f
1 parent abaa9e6
commit 41a404f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/plugins/intel_gpu/src/plugin/ops/transpose.cpp b/src/plugins/intel_gpu/src/plugin/ops/transpose.cpp
@@ -46,8 +46,11 @@ static void CreateTransposeOp(Program& p, const std::shared_ptr<ngraph::op::v1::
     // Handle Transpose operation related to ConvertColor operation:
     // In case of ConvertColor operation we have NHWC (byxf) input format which should be converted to
     // NCHW (bfyx) by this Permute, so we replace Permute with Reorder (to bfyx) primitve
-    auto input = op->get_input_node_shared_ptr(0);
-    if (is_convert_color_type(input) && order == std::vector<uint16_t>{0, 3, 1, 2}) {
+    auto input = op->get_input_size() > 0 ? op->get_input_node_shared_ptr(0) : nullptr;
+    // Handle the case ConvertColor -> FakeQuantize -> Permute
+    auto input1 = input ? (input->get_input_size() > 0 ? input->get_input_node_shared_ptr(0) : nullptr) : nullptr;
+    if (((input && is_convert_color_type(input)) || (input1 && is_convert_color_type(input1)))
+            && order == std::vector<uint16_t>{0, 3, 1, 2}) {
         auto precision = input->get_element_type();
         p.AddPrimitive(cldnn::reorder(layerName,
                                       inputPrimitives[0],