TopK constant provenance (#3658)

68472cc9 · Scott Cyphers · GitHub · 669532d8 · 68472cc9
Unverified Commit 68472cc9 authored Sep 24, 2019 by Scott Cyphers Committed by GitHub Sep 24, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

topk.cpp src/ngraph/op/topk.cpp +1 -0

No files found.
--- a/src/ngraph/op/topk.cpp
+++ b/src/ngraph/op/topk.cpp
@@ -42,6 +42,7 @@ op::TopK::TopK(const Output<Node>& arg,
    , m_compute_max(compute_max)
    , m_sort(sort)
 {
+    add_provenance_group_member(input(1).get_source_output().get_node_shared_ptr());
    constructor_validate_and_infer_types();
 }