Commit 2a150ccc authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add an onnx tanh test.

parent abf898da
...@@ -221,6 +221,18 @@ void cosh_test() ...@@ -221,6 +221,18 @@ void cosh_test()
EXPECT(p == prog); EXPECT(p == prog);
} }
void tanh_test()
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1}});
p.add_instruction(migraphx::op::tanh{}, input);
auto prog = migraphx::parse_onnx("tanh_test.onnx");
EXPECT(p == prog);
}
void asin_test() void asin_test()
{ {
migraphx::program p; migraphx::program p;
...@@ -271,6 +283,7 @@ int main() ...@@ -271,6 +283,7 @@ int main()
tan_test(); tan_test();
sinh_test(); sinh_test();
cosh_test(); cosh_test();
tanh_test();
asin_test(); asin_test();
acos_test(); acos_test();
atan_test(); atan_test();
......
 tanh-example:;
xy"Tanh test_tanhZ
x

b
y

B
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment