From 4476a66f338a8a6ba857d2e1c98441633f6acbad Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Mon, 2 Oct 2023 20:43:23 -0700 Subject: [PATCH] Update js_execution_provider.cc --- onnxruntime/core/providers/js/js_execution_provider.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/onnxruntime/core/providers/js/js_execution_provider.cc b/onnxruntime/core/providers/js/js_execution_provider.cc index f29a348cd05c5..b85f97ca857d4 100644 --- a/onnxruntime/core/providers/js/js_execution_provider.cc +++ b/onnxruntime/core/providers/js/js_execution_provider.cc @@ -659,7 +659,8 @@ std::vector> JsExecutionProvider::GetCapabili continue; } candidates.push_back(node.Index()); - if(node.Domain() != 'com.ms.internal.nhwc'){ + // If the node was added by layout transformer, do not move it to CPU + if(node.Domain() != "com.ms.internal.nhwc"){ tenative_candidates.push_back(node.Index()); } }