From 53038b33edb45eec059f4a89d44898ca37e1db26 Mon Sep 17 00:00:00 2001 From: jignparm Date: Wed, 10 Apr 2019 18:52:10 -0700 Subject: [PATCH] BuildFusedKernelDef uses N^2 algorithm verifying input constraints; session load time is huge for fused nodes (#804) This optimization is required for WinML to prevent unit test time out. --- onnxruntime/core/framework/graph_partitioner.cc | 4 ---- 1 file changed, 4 deletions(-) diff --git a/onnxruntime/core/framework/graph_partitioner.cc b/onnxruntime/core/framework/graph_partitioner.cc index 367bd4c334..003bc060f0 100644 --- a/onnxruntime/core/framework/graph_partitioner.cc +++ b/onnxruntime/core/framework/graph_partitioner.cc @@ -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; }