We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c78ee80 commit 3daa5b5Copy full SHA for 3daa5b5
nnvm/python/nnvm/frontend/onnx.py
@@ -307,10 +307,7 @@ class Prelu(OnnxOpConverter):
307
def _impl_v1(cls, inputs, attr, params):
308
assert len(inputs) == 2, "Prelu need 2 inputs, {} given".format(
309
len(inputs))
310
- channels = infer_channels(inputs[1], params, False)
311
- if channels == 1:
312
- return inputs[0] * inputs[1]
313
- return _sym.broadcast_mul(inputs[0], inputs[1])
+ return _sym.prelu(inputs[0], inputs[1])
314
315
316
class Reciprocal(OnnxOpConverter):
0 commit comments