Commit 82b0a072 authored by Khalique's avatar Khalique
Browse files

added dropout onnx file

parent 70a993f3
...@@ -158,6 +158,17 @@ void transpose_test() ...@@ -158,6 +158,17 @@ void transpose_test()
EXPECT(p == prog); EXPECT(p == prog);
} }
void dropout_test()
{
migraph::program p;
auto input = p.add_parameter("0", migraph::shape{migraph::shape::float_type, {1, 3, 2, 2}});
p.add_instruction(migraph::op::identity{}, input);
auto prog = migraph::parse_onnx("dropout_test.onnx");
EXPECT(p == prog);
}
int main() int main()
{ {
pytorch_conv_bias_test(); pytorch_conv_bias_test();
...@@ -169,4 +180,5 @@ int main() ...@@ -169,4 +180,5 @@ int main()
globalavgpool_test(); globalavgpool_test();
globalmaxpool_test(); globalmaxpool_test();
transpose_test(); transpose_test();
dropout_test();
} }
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