From 8cd959db8f21cd62efa137268c88cc3bbe2ac441 Mon Sep 17 00:00:00 2001
From: vandanavk <vandana268@gmail.com>
Date: Tue, 20 Nov 2018 14:08:37 -0800
Subject: [PATCH] Add Flatten before Gemm

---
 python/mxnet/contrib/onnx/mx2onnx/_op_translations.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py b/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py
index 73ca07be76ee..65ca2c25b2f2 100644
--- a/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py
+++ b/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py
@@ -232,6 +232,17 @@ def convert_fully_connected(node, **kwargs):
 
     fcnode = []
 
+    op_name = "flatten_" + str(kwargs["idx"])
+    flatten_node = onnx.helper.make_node(
+        'Flatten',
+        inputs=[input_nodes[0]],
+        outputs=[op_name],
+        name=op_name
+    )
+
+    input_nodes[0] = op_name
+    fcnode.append(flatten_node)
+
     if no_bias:
         data_type = onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[np.dtype('int64')]
         bias_name = "bias" + str(kwargs["idx"])