Fixing unity substituion tests

Author: Victor Li

lib/substitutions/include/substitutions/unity_substitution_set.h

@@ -21,7 +21,6 @@ Substitution create_replicate_linear_combine(nonnegative_int num_dims,
                                              bool use_bias);
 Substitution create_partition_linear_combine(nonnegative_int num_dims,
                                              nonnegative_int degree,
-                                             Activation activation,
                                              bool use_bias);
 Substitution create_partition_conv2d_combine(nonnegative_int num_dims,
                                              nonnegative_int degree);

lib/substitutions/src/substitutions/unity_substitution_set.cc

@@ -21,6 +21,10 @@ std::vector<Substitution>
           create_replicate_linear_combine(num_dims, degree, true));
       substitutions.push_back(
           create_replicate_linear_combine(num_dims, degree, false));
+      substitutions.push_back(
+          create_partition_linear_combine(num_dims, degree, true));
+      substitutions.push_back(
+          create_partition_linear_combine(num_dims, degree, false));
     }
   }
   substitutions.push_back(create_fuse_linear_activation(Activation::RELU));
@@ -63,7 +67,6 @@ Substitution create_replicate_linear_combine(nonnegative_int num_dims,
       op_type_equals_constraint(OperatorType::LINEAR),
       op_attr_key_equals(OperatorAttributeKey::BIAS,
                          OperatorAttributeValue{use_bias}),
-
       op_attr_key_divisible_by(OperatorAttributeKey::OUT_CHANNELS, degree),
   }};
 
@@ -146,9 +149,7 @@ Substitution create_replicate_linear_combine(nonnegative_int num_dims,
 
 Substitution create_partition_linear_combine(nonnegative_int num_dims,
                                              nonnegative_int degree,
-                                             Activation activation,
                                              bool use_bias) {
-
   SubstitutionBuilder b;
 
   auto [p_input, o_input] = b.add_input(tensor_attribute_pattern_match_all());