tensorflow-onnx icon indicating copy to clipboard operation
tensorflow-onnx copied to clipboard

Convert Dense to "FC" instead of "GEMM"

Open aviaisr opened this issue 3 years ago • 1 comments

Is there a way to convert force the conversion of tf.Densse layer to MatMul + Add instead of GEMM? I'm converting onnx to DLC and GEMM is causing problems.

aviaisr avatar Jan 31 '22 09:01 aviaisr

This was a gemm_rewriter to rewriter the MatMul + Add op to GEMM . https://github.com/onnx/tensorflow-onnx/blob/master/tf2onnx/rewriter/gemm_rewriter.py.

You can just remove the related code from this PR add gemm_rewriter and the corresponding test

By the way, is the GEMM op causing which problems?

hwangdeyu avatar Feb 08 '22 12:02 hwangdeyu

It's been over 3 months, so closing this. Feel free to open a new one if the issue still exists.

fatcat-z avatar Oct 11 '22 05:10 fatcat-z