Skip to content

Commit da17e48

Browse files
Thomas NoelTalmaj
authored andcommitted
Added GreaterOrEqual operation node type
1 parent 03e81a3 commit da17e48

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

onnx2pytorch/convert/operations.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,8 @@ def convert_operations(onnx_graph, opset_version, batch_dim=0, enable_pruning=Tr
131131
op = GlobalAveragePool()
132132
elif node.op_type == "Greater":
133133
op = OperatorWrapper(torch.greater)
134+
elif node.op_type == "GreaterOrEqual":
135+
op = OperatorWrapper(torch.greater_equal)
134136
elif node.op_type == "Identity":
135137
op = nn.Identity()
136138
elif node.op_type == "InstanceNormalization":

0 commit comments

Comments
 (0)