outputs to 3

microsoft · Jan 24, 2024 · 7456de4 · 7456de4
1 parent f5c9a41
commit 7456de4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orttraining/orttraining/test/gradient/gradient_ops_test.cc b/orttraining/orttraining/test/gradient/gradient_ops_test.cc
@@ -700,7 +700,7 @@ TEST(GradientCheckerTest, SplitGrad) {
 
   ASSERT_STATUS_OK(gradient_checker.ComputeGradientError(op_def, {shape}, {{3, 5}, {3, 5}, {3, 5}}, &max_error,
                                                          {MakeAttribute("axis", int64_t(0)),
-                                                          MakeAttribute("num_outputs", int64_t(2))}));
+                                                          MakeAttribute("num_outputs", int64_t(3))}));
   EXPECT_IS_TINY(max_error);
 
   // opset13 test