Skip to content
Snippets Groups Projects
Commit 669b44c1 authored by Tatsuya Nishiyama's avatar Tatsuya Nishiyama Committed by Tianqi Chen
Browse files

Fix the gemm conversion in onnx frontend (#1241)

parent 552fa71b
No related branches found
No related tags found
No related merge requests found
......@@ -186,6 +186,7 @@ class Gemm(OnnxOpConverter):
inputs[0] = _sym.transpose(inputs[0], axes=(1, 0))
if not transB:
inputs[1] = _sym.transpose(inputs[1], axes=(1, 0))
inputs[0] = _sym.flatten(inputs[0])
return _sym.dense(
alpha * inputs[0], inputs[1], beta * inputs[2], units=channels)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment