Add missing input (#3691)

* Add missing input

* If builder does nothing, don't dive into the inputs

* Reverse test

* simplify, add test

* Add test

src/ngraph/builder/quantization.cpp

@@ -440,8 +440,15 @@ namespace ngraph
             }
             return make_shared<op::QuantizedDotBias>(
                        input, filters, mybias, requantization_scale, requantize, with_relu)
-                ->add_provenance_group_members_above(
-                    {min_input, max_input, min_filter, max_filter, min_output, max_output});
+                ->add_provenance_group_members_above({input,
+                                                      filters,
+                                                      bias,
+                                                      min_input,
+                                                      max_input,
+                                                      min_filter,
+                                                      max_filter,
+                                                      min_output,
+                                                      max_output});
         }
 
         shared_ptr<Node> ScaledQuantizedDot(const Output<Node>& input,

src/ngraph/node.cpp

@@ -328,7 +328,13 @@ shared_ptr<Node> Node::add_provenance_group_members_above(const OutputVector& ba
     set<Node*> base_set;
     for (auto& output : base)
     {
-        base_set.insert(output.get_node());
+        Node* node = output.get_node();
+        if (node == this)
+        {
+            // A builder did nothing
+            return shared_from_this();
+        }
+        base_set.insert(node);
     }
     vector<Node*> todo;
     for (auto input : inputs())

test/provenance.cpp

@@ -431,3 +431,24 @@ TEST(provenance, fused)
         }
     });
 }
+
+TEST(provenance, empty_group)
+{
+    auto p1 = make_shared<op::Parameter>(element::i32, PartialShape{2, 3, 4});
+    p1->add_provenance_tag("P1");
+    auto abs = make_shared<op::Abs>(p1);
+    // Make sure group is empty
+    abs->add_provenance_group_members_above({abs});
+    abs->add_provenance_tag("abs");
+    for (auto node : topological_sort(NodeVector{abs}))
+    {
+        if (node == p1)
+        {
+            EXPECT_EQ(node->get_provenance_tags(), (ProvSet{"P1"}));
+        }
+        else
+        {
+            EXPECT_EQ(node->get_provenance_tags(), (ProvSet{"abs"}));
+        }
+    }
+}