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

bug fix for symbolic shape infer (#13067)

Browse source
This commit is contained in:
donglinb 2022-10-15 05:06:31 +08:00 committed by GitHub
parent 65c67764ae
commit c4a52820a5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -41,6 +41,13 @@ def get_shape_from_type_proto(type_proto):
return None # note no shape is different from shape without dim (scalar)
def get_elem_type_from_type_proto(type_proto):
if is_sequence(type_proto):
return type_proto.sequence_type.elem_type.tensor_type.elem_type
else:
return type_proto.tensor_type.elem_type
def get_shape_from_value_info(vi):
cls_type = vi.type.WhichOneof("value")
if cls_type is None:
@ -570,7 +577,7 @@ class SymbolicShapeInference:
vi.CopyFrom(
helper.make_tensor_value_info(
node.output[0],
self.known_vi_[node.input[0]].type.tensor_type.elem_type,
get_elem_type_from_type_proto(self.known_vi_[node.input[0]].type),
self._get_shape(node, 0),
)
)