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

Only fuse when output count of add is 1

Browse source
This commit is contained in:
Yufeng Li 2020-01-21 15:34:16 -08:00 committed by Ryan Hill
parent 5ecabe1b15
commit 1a5a25331d
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/optimizer/skip_layer_norm_fusion.cc
View file

@ -25,7 +25,8 @@ static bool IsSupportedDataType(const Node& node) {
static bool CheckFirstAdd(Node& add, ProviderType providertype) {
if (providertype != add.GetExecutionProviderType() ||
!IsSupportedDataType(add)) {
!IsSupportedDataType(add) ||
add.GetOutputEdgesCount() != 1) {
return false;
}
@ -58,7 +59,8 @@ static bool CheckFirstAdd(Node& add, ProviderType providertype) {
// The 2nd input should be a 1D constant value
static bool CheckSecondAdd(Node& add, ProviderType providertype) {
if (providertype != add.GetExecutionProviderType() ||
!IsSupportedDataType(add)) {
!IsSupportedDataType(add) ||
add.GetOutputEdgesCount() != 1) {
return false;
}