saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-20 19:12:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use 'as_scalar' when checking the 'cond' value of 'If' (#7063)

Browse source 
#6884
This commit is contained in:
Scott McKay 2021-03-22 18:04:38 +10:00 committed by GitHub
parent cec919bae9
commit b2c6617b0f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/symbolic_shape_infer.py
View file

@ -746,7 +746,7 @@ class SymbolicShapeInference:
subgraphs = [get_attribute(node, 'then_branch'), get_attribute(node, 'else_branch')]
cond = self._try_get_value(node, 0)
if cond is not None:
if cond > 0:
if as_scalar(cond) > 0:
subgraphs[1].CopyFrom(subgraphs[0])
else:
subgraphs[0].CopyFrom(subgraphs[1])