so cool

AndrewZhaoLuo · Aug 31, 2021 · 2a7b2b7 · 2a7b2b7
1 parent 86229c2
commit 2a7b2b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -3532,7 +3532,7 @@ def _impl_v13(cls, inputs, attr, params):
 
         if reduction == "mean":
             return relay.sum(loss) / weight_total
-        elif reduction == "sum":
+        if reduction == "sum":
             return relay.sum(loss)
         # Case reduction == 'none'
         return loss