Add provenance for ops constructed in op constructors. (#3663)

src/ngraph/op/experimental/generate_mask.cpp

@@ -54,6 +54,10 @@ op::GenerateMask::GenerateMask(const Output<Node>& training,
                  make_shared<op::Constant>(element::i32, Shape{}, std::vector<int32_t>{use_seed}));
     set_argument(3, make_shared<op::Constant>(element::u64, Shape{}, std::vector<uint64_t>{seed}));
     set_argument(4, make_shared<op::Constant>(element::f64, Shape{}, std::vector<double>{prob}));
+    add_provenance_group_member(input_value(1).get_node_shared_ptr());
+    add_provenance_group_member(input_value(2).get_node_shared_ptr());
+    add_provenance_group_member(input_value(3).get_node_shared_ptr());
+    add_provenance_group_member(input_value(4).get_node_shared_ptr());
     constructor_validate_and_infer_types();
 }
 

src/ngraph/op/lrn.cpp

@@ -26,6 +26,7 @@ const string op::LRN::type_name{"LRN"};
 op::LRN::LRN(const Output<Node>& arg, double alpha, double beta, double bias, size_t size)
     : LRN(arg, op::Constant::create(element::i64, Shape{1}, {1}), alpha, beta, bias, size)
 {
+    add_provenance_group_member(input_value(1).get_node_shared_ptr());
 }
 
 op::LRN::LRN(const Output<Node>& arg,

src/ngraph/op/topk.cpp

@@ -38,6 +38,7 @@ op::TopK::TopK(const Output<Node>& arg,
     , m_compute_max(compute_max)
     , m_sort(sort)
 {
+    add_provenance_group_member(input_value(1).get_node_shared_ptr());
     constructor_validate_and_infer_types();
 }