diff options
Diffstat (limited to 'tensorflow/core/ops/nn_ops_test.cc')
-rw-r--r-- | tensorflow/core/ops/nn_ops_test.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/ops/nn_ops_test.cc b/tensorflow/core/ops/nn_ops_test.cc index a60b1c3788..51e4f8bffe 100644 --- a/tensorflow/core/ops/nn_ops_test.cc +++ b/tensorflow/core/ops/nn_ops_test.cc @@ -412,7 +412,8 @@ TEST(NNOpsTest, Dilation2DBackpropFilter_ShapeFn) { TEST(NNOpsTest, MergeBothInputs_ShapeFn) { for (const char* op_name : - {"ReluGrad", "Relu6Grad", "EluGrad", "SoftplusGrad", "SoftsignGrad"}) { + {"ReluGrad", "Relu6Grad", "EluGrad", "SeluGrad", "SoftplusGrad", + "SoftsignGrad"}) { ShapeInferenceTestOp op(op_name); INFER_OK(op, "?;?", "in0|in1"); |