search

NNgen / nngen

NNgen: A Fully-Customizable Hardware Synthesis Compiler for Deep Neural Network
Apache License 2.0
336 stars 46 forks source link

Fix onnx/gemm.py to pass the tests/onnx_matrix_linear. #57

Open hirayamy opened 1 year ago

hirayamy commented 1 year ago

I got the AttirbuteError in Gemm function when the src_node of gemm is a placeholder, such as inputs. So I modified the gemm function to avoid this attribution error by skipping the 'op_type' reference to NoneType objects.