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

apache · Sep 1, 2019 · c1c7b9b · c1c7b9b
1 parent d08c74c
commit c1c7b9b
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/nnvm/python/nnvm/frontend/onnx.py b/nnvm/python/nnvm/frontend/onnx.py
@@ -307,10 +307,7 @@ class Prelu(OnnxOpConverter):
     def _impl_v1(cls, inputs, attr, params):
         assert len(inputs) == 2, "Prelu need 2 inputs, {} given".format(
             len(inputs))
-        channels = infer_channels(inputs[1], params, False)
-        if channels == 1:
-            return inputs[0] * inputs[1]
-        return _sym.broadcast_mul(inputs[0], inputs[1])
+        return _sym.prelu(inputs[0], inputs[1])
 
 
 class Reciprocal(OnnxOpConverter):