From 51295b26eddffa5069cf58abeb634b8e19a7a89a Mon Sep 17 00:00:00 2001 From: lucylq Date: Fri, 6 Jun 2025 17:08:22 -0700 Subject: [PATCH] Revert "Make op_glu_test input asymmetric (#11294)" This reverts commit 0c9a4f5f476a7ff796245918b03f026c22c184d6. --- kernels/test/op_glu_test.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/kernels/test/op_glu_test.cpp b/kernels/test/op_glu_test.cpp index b18117eaa4e..f8bf22dae63 100644 --- a/kernels/test/op_glu_test.cpp +++ b/kernels/test/op_glu_test.cpp @@ -51,20 +51,20 @@ class OpGluOutTest : public OperatorTest { const std::vector out_sizes_1 = {2, 2}; // Valid input should give the expected output - Tensor in = tf.make(sizes, {0, 1, 2, 3, 4, 5, 6, 7}); + Tensor in = tf.ones(sizes); Tensor out = tf_out.zeros(out_sizes_1); op_glu_out(in, 0, out); - expect_tensor_close( + expect_tensor_close( out, tf_out.make( - out_sizes_1, /*data=*/{0, 0.99330717, 1.99505484, 2.99726701})); + out_sizes_1, /*data=*/{0.731059, 0.731059, 0.731059, 0.731059})); const std::vector out_sizes_2 = {4, 1}; out = tf_out.zeros(out_sizes_2); op_glu_out(in, 1, out); - expect_tensor_close( + expect_tensor_close( out, tf_out.make( - out_sizes_2, /*data=*/{0, 1.90514827, 3.97322869, 5.99453402})); + out_sizes_2, /*data=*/{0.731059, 0.731059, 0.731059, 0.731059})); } // Mismatched shape tests.