IntelGPU backend: Fix reshape operation (#2201)

24bd105f · Sergey Shalnov · Scott Cyphers · 31210402 · 24bd105f
Commit 24bd105f authored Dec 11, 2018 by Sergey Shalnov Committed by Scott Cyphers Dec 11, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

intelgpu_backend.cpp src/ngraph/runtime/intelgpu/intelgpu_backend.cpp +4 -1

No files found.
--- a/src/ngraph/runtime/intelgpu/intelgpu_backend.cpp
+++ b/src/ngraph/runtime/intelgpu/intelgpu_backend.cpp
@@ -926,7 +926,10 @@ runtime::Handle runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function>
            }
            else
            {
-                do_equal_propagation(topology, get_input_name(op), get_output_name(op));
+                const cldnn::tensor new_shape =
+                    intelgpu_space::create_cldnn_tensor(get_output_shape(op));
+                const cldnn::reshape reshape_op(get_output_name(op), get_input_name(op), new_shape);
+                topology.add(reshape_op);
            }
            break;
        }