Skip to content
Snippets Groups Projects
Commit cf27e975 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Fixed Gemm support

parent ce09ba57
No related branches found
No related tags found
No related merge requests found
......@@ -116,6 +116,7 @@ class DFG(object):
"MaxPool": g.MaxPool2DNode,
"AveragePool": g.AveragePool2DNode,
"MatMul": g.MatMulNode,
"Gemm": g.MatMulNode,
"Add": g.AddNode,
"Softmax": g.SoftMaxNode,
"Relu": g.ReluNode,
......
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