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

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

Browse Source 
Co-authored-by: Tang Wei <wei1.tang@intel.com>

Co-authored-by: Tang Wei <wei1.tang@intel.com>
This commit is contained in:
Kurt ChenandTang Wei authored and GitHub committed 2022-08-25 11:54:55 +04:00
1 parent f994e307a9
commit febead7e8c
1 file changed
+5 -2
Download Patch File Download Diff File Expand all files Collapse all files
src/plugins/intel_gpu/src/plugin/ops/transpose.cpp
+5 -2
View File
@@ -45,8 +45,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],