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

BuildFusedKernelDef uses N^2 algorithm verifying input constraints; session load time is huge for fused nodes (#804)

Browse source 
This optimization is required for WinML to prevent unit test time out.
This commit is contained in:
jignparm 2019-04-10 18:52:10 -07:00 committed by Ke Zhang
parent d17ae5c093
commit 53038b33ed
1 changed files with 0 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/framework/graph_partitioner.cc
View file

@ -48,10 +48,6 @@ KernelDefBuilder& BuildFusedKernelDef(KernelDefBuilder& builder, const onnxrunti
.SetDomain(schema->domain())
.SinceVersion(schema->SinceVersion())
.Provider(node.GetExecutionProviderType());
auto& inputs = node.InputDefs();
for (auto input : inputs) {
builder.TypeConstraint(input->Name(), DataTypeImpl::TypeFromProto(*input->TypeAsProto()));
}
return builder;
}