Skip to content

T
tic
Commit c1c7b9b1 by Alexey Golunov Committed by Jared Roesch
Options

[NNVM][FRONTEND][ONNX] Fix PReLU conversion (#3813)

parent d08c74ca
Showing with 1 additions and 4 deletions
nnvm/python/nnvm/frontend/onnx.py
...@@ -307,10 +307,7 @@ class Prelu(OnnxOpConverter): ...@@ -307,10 +307,7 @@ class Prelu(OnnxOpConverter):
def _impl_v1(cls, inputs, attr, params): def _impl_v1(cls, inputs, attr, params):
assert len(inputs) == 2, "Prelu need 2 inputs, {} given".format( assert len(inputs) == 2, "Prelu need 2 inputs, {} given".format(
len(inputs)) len(inputs))
channels = infer_channels(inputs[1], params, False) return _sym.prelu(inputs[0], inputs[1])
if channels == 1:
return inputs[0] * inputs[1]
return _sym.broadcast_mul(inputs[0], inputs[1])
class Reciprocal(OnnxOpConverter): class Reciprocal(OnnxOpConverter):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment