TopK constant provenance (#3658)

NervanaSystems · Sep 24, 2019 · 68472cc · 68472cc
1 parent 669532d
commit 68472cc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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();
 }