Skip to content

Commit 51295b2

Browse files
committed
Revert "Make op_glu_test input asymmetric (#11294)"
This reverts commit 0c9a4f5.
1 parent e5b8735 commit 51295b2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

kernels/test/op_glu_test.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,20 +51,20 @@ class OpGluOutTest : public OperatorTest {
5151
const std::vector<int32_t> out_sizes_1 = {2, 2};
5252

5353
// Valid input should give the expected output
54-
Tensor in = tf.make(sizes, {0, 1, 2, 3, 4, 5, 6, 7});
54+
Tensor in = tf.ones(sizes);
5555
Tensor out = tf_out.zeros(out_sizes_1);
5656
op_glu_out(in, 0, out);
57-
expect_tensor_close<OUT_DTYPE>(
57+
expect_tensor_close<DTYPE>(
5858
out,
5959
tf_out.make(
60-
out_sizes_1, /*data=*/{0, 0.99330717, 1.99505484, 2.99726701}));
60+
out_sizes_1, /*data=*/{0.731059, 0.731059, 0.731059, 0.731059}));
6161
const std::vector<int32_t> out_sizes_2 = {4, 1};
6262
out = tf_out.zeros(out_sizes_2);
6363
op_glu_out(in, 1, out);
64-
expect_tensor_close<OUT_DTYPE>(
64+
expect_tensor_close<DTYPE>(
6565
out,
6666
tf_out.make(
67-
out_sizes_2, /*data=*/{0, 1.90514827, 3.97322869, 5.99453402}));
67+
out_sizes_2, /*data=*/{0.731059, 0.731059, 0.731059, 0.731059}));
6868
}
6969

7070
// Mismatched shape tests.

0 commit comments

Comments
 (0)