Fix a bug.

46f4acae · amy.zhuang · 0e8be7e3 · 46f4acae
Commit 46f4acae authored Oct 19, 2018 by amy.zhuang
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 3 deletions

cpu_external_function.cpp src/ngraph/runtime/cpu/cpu_external_function.cpp +10 -3

No files found.
--- a/src/ngraph/runtime/cpu/cpu_external_function.cpp
+++ b/src/ngraph/runtime/cpu/cpu_external_function.cpp
@@ -1172,10 +1172,17 @@ void runtime::cpu::CPU_ExternalFunction::process_in_place_concat(
                    bool found_last_concat = true;
                    for (auto user : concat->get_users())
                    {
-                        if (dynamic_pointer_cast<ngraph::op::Concat>(user))
+                        if (auto user_concat = dynamic_pointer_cast<ngraph::op::Concat>(user))
                        {
-                            found_last_concat = false;
+                            if (auto op_annotations = user_concat->get_op_annotations())
-                            break;
+                            {
+                                auto in_place_oi_pairs = op_annotations->get_in_place_oi_pairs();
+                                if (in_place_oi_pairs.size() > 0)
+                                {
+                                    found_last_concat = false;
+                                    break;
+                                }
+                            }
                        }
                    }