PR feedback

* Removed exported jsepGetMLContext
* Fixed check for ml-tensor and ml-location when creating an MLContext

Author: egalli

js/web/lib/wasm/wasm-types.ts

@@ -138,12 +138,6 @@ export declare namespace JSEP {
      * @returns
      */
     jsepRegisterMLContext: (sessionId: number, context: MLContext) => void;
-    /**
-     * [exported from pre-jsep.js] Get MLContext for a session.
-     * @param sessionId - specify the session ID.
-     * @returns the MLContext.
-     */
-    jsepGetMLContext: (sessionId: number) => MLContext;
     /**
      * [exported from pre-jsep.js] Reserve a MLBuffer ID attached to the current session.
      * @returns the MLBuffer ID.

js/web/test/test-runner.ts

@@ -257,7 +257,7 @@ export class ModelTestContext {
       const executionProviderConfig =
           modelTest.backend === 'webnn' ? (testOptions?.webnnOptions || {name: 'webnn'}) : modelTest.backend!;
       let mlContext: MLContext|undefined;
-      if (modelTest.ioBinding.includes('ml-tensor') || modelTest.ioBinding.includes('ml-location')) {
+      if(['ml-tensor', 'ml-location'].includes(modelTest.ioBinding)) {
 
         const webnnOptions = executionProviderConfig as ort.InferenceSession.WebNNExecutionProviderOption;
         const deviceType = (webnnOptions as ort.InferenceSession.WebNNContextOptions)?.deviceType;

onnxruntime/wasm/pre-jsep.js

@@ -226,9 +226,6 @@ Module['jsepInit'] = (name, params) => {
       backend['onReleaseSession'](sessionId);
     };
     Module['jsepReleaseBufferId'] = Module.jsepReleaseBufferId;
-    Module['jsepGetMLContext'] = sessionId => {
-      return backend['getMLContext'](sessionId);
-    };
     Module['jsepCreateMLBufferDownloader'] = (bufferId, type) => {
       return backend['createMLBufferDownloader'](bufferId, type);
     }