oramasearch / onnx-go

onnx-go gives the ability to import a pre-trained neural network within Go without being linked to a framework or library.
https://blog.owulveryck.info/2019/04/03/from-a-project-to-a-product-the-state-of-onnx-go.html
MIT License
704 stars 72 forks source link

Implement operator PRelu for backend Gorgonia #177

Closed kafan1986 closed 4 years ago

kafan1986 commented 4 years ago

Why is this operator needed?

To run resnet34

Implementation

https://github.com/onnx/onnx/blob/master/docs/Operators.md#PRelu

Given that it is to run a pretrained model. It should be exactly similar to leakyRelu.

Link to existing material on the backend

onnx: operator PRelu not implemented

Tests

<-- Give any information to test the operator go test -run=ONNX/TestOperator -->