code generation for dropout layer

lcskrishna · lcskrishna · commit cd636c5480c6 · 2017-12-21T13:10:31.000-08:00
diff --git a/src/onnx_gdf_parser.cpp b/src/onnx_gdf_parser.cpp
@@ -338,6 +338,9 @@ int writeGDF
 			ofsGDF << "node org.khronos.nn_extension.convolution_layer " << layer_input << " " << layer_weights << " " << layer_bias
 				<< " " << layer_output << "_params" << " " << layer_output << std::endl; 	
 		}
+		else if(layer_type == "Dropout") {
+			ofsGDF << "node org.khronos.openvx.copy " << layer_input << " " << layer_output << std::endl;
+		}
 		
 		if(i == net.size() - 1) {
 			ofsGDF << "write " << layer_output << " output.f32" << std::endl;

Original file line number	Diff line number	Diff line change
`@@ -338,6 +338,9 @@ int writeGDF`
`338`	`338`	`ofsGDF << "node org.khronos.nn_extension.convolution_layer " << layer_input << " " << layer_weights << " " << layer_bias`
`339`	`339`	`<< " " << layer_output << "_params" << " " << layer_output << std::endl;`
`340`	`340`	`}`
	`341`	`+ else if(layer_type == "Dropout") {`
	`342`	`+ ofsGDF << "node org.khronos.openvx.copy " << layer_input << " " << layer_output << std::endl;`
	`343`	`+ }`
`341`	`344`
`342`	`345`	`if(i == net.size() - 1) {`
`343`	`346`	`ofsGDF << "write " << layer_output << " output.f32" << std::endl;`