Update js_execution_provider.cc

snnn · Oct 3, 2023 · 4476a66 · 4476a66
1 parent d646423
commit 4476a66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/js/js_execution_provider.cc b/onnxruntime/core/providers/js/js_execution_provider.cc
@@ -659,7 +659,8 @@ std::vector<std::unique_ptr<ComputeCapability>> 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());
     }
   }