Merge pull request #110 from ROCmSoftwarePlatform/dropout

Dropout

src/onnx/onnx.cpp

@@ -56,6 +56,8 @@ struct onnx_parser
         add_generic_op("Relu", op::relu{});
         add_generic_op("Sub", op::sub{});
         add_generic_op("Sum", op::add{});
+        // disable dropout for inference
+        add_generic_op("Dropout", op::identity{});
 
         add_mem_op("ImageScaler", &onnx_parser::parse_imagescaler);
         add_mem_op("LeakyRelu", &onnx_parser::parse_leaky_relu);

test/onnx/dropout_test.onnx

+dropout-example:Y
+
+
0
1"Dropouttest-dropoutZ
+
0
+

+

+
+
+b
+
1
+

+

+
+
+B
\ No newline at end of file

test/onnx/onnx_test.cpp

@@ -158,6 +158,17 @@ void transpose_test()
     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()
 {
     pytorch_conv_bias_test();
@@ -169,4 +180,5 @@ int main()
     globalavgpool_test();
     globalmaxpool_test();
     transpose_test();
+    dropout_test();
 }