aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/tests/randomized_tests.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/tests/randomized_tests.cc')
-rw-r--r--tensorflow/compiler/tests/randomized_tests.cc17
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/randomized_tests.cc b/tensorflow/compiler/tests/randomized_tests.cc
index d3821ad02e..825fd9de2e 100644
--- a/tensorflow/compiler/tests/randomized_tests.cc
+++ b/tensorflow/compiler/tests/randomized_tests.cc
@@ -1434,6 +1434,23 @@ TEST_F(OpTest, EluGrad) {
});
}
+TEST_F(OpTest, Selu) {
+ Repeatedly([this]() {
+ return ExpectTfAndXlaOutputsAreClose(
+ OpTestBuilder("Selu").RandomInput(DT_FLOAT).Attr("T", DT_FLOAT));
+ });
+}
+
+TEST_F(OpTest, SeluGrad) {
+ Repeatedly([this]() {
+ auto dims = RandomDims();
+ return ExpectTfAndXlaOutputsAreClose(OpTestBuilder("SeluGrad")
+ .RandomInput(DT_FLOAT, dims)
+ .RandomInput(DT_FLOAT, dims)
+ .Attr("T", DT_FLOAT));
+ });
+}
+
TEST_F(OpTest, Equal) {
Repeatedly([this]() {
DataType type = Choose<DataType>({DT_INT32, DT_FLOAT});