From 7768a0737386d5d48bb93de7d329df13c30c6da3 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Mon, 2 Jun 2025 12:38:11 -0700 Subject: [PATCH] Update [ghstack-poisoned] --- 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 f8bf22dae63..b18117eaa4e 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.ones(sizes); + Tensor in = tf.make(sizes, {0, 1, 2, 3, 4, 5, 6, 7}); 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.731059, 0.731059, 0.731059, 0.731059})); + out_sizes_1, /*data=*/{0, 0.99330717, 1.99505484, 2.99726701})); 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.731059, 0.731059, 0.731059, 0.731059})); + out_sizes_2, /*data=*/{0, 1.90514827, 3.97322869, 5.99453402})); } // Mismatched shape tests.