Copy RT info for parameters (#3969)

d6c692af · Ilya Churaev · Scott Cyphers · 3d1335fe · d6c692af
Commit d6c692af authored Dec 02, 2019 by Ilya Churaev Committed by Scott Cyphers Dec 02, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

specialize_function.cpp src/ngraph/specialize_function.cpp +1 -0

No files found.
--- a/src/ngraph/specialize_function.cpp
+++ b/src/ngraph/specialize_function.cpp
@@ -60,6 +60,7 @@ std::shared_ptr<Function>
            m[f->get_parameters()[i].get()] =
                std::make_shared<op::Parameter>(parameter_element_types[i], parameter_shapes[i]);
        }
+        m[f->get_parameters()[i].get()]->get_rt_info() = f->get_parameters()[i]->get_rt_info();
    }

    for (auto old_node : f->get_ordered_ops())