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

Fix a bug in fusion_embedlayer.py (#10022)

Browse source
This commit is contained in:
Ye Wang 2021-12-14 12:50:35 -08:00 committed by GitHub
parent 5be0fa13c0
commit 703becd796
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/transformers/fusion_embedlayer.py
View file

@ -116,7 +116,7 @@ class FusionEmbedLayerNoMask(Fusion):
if path1 is None:
return False
expand, shape = path1
expand, shape = path1[0], path1[-1]
if shape.input[0] != input_ids:
return False