Function resetBackend

e2e/benchmarks/benchmark_util.js:661–684

* Reset the target backend. * * @param backendName The name of the backend to be reset.

(backendName)

 * @param backendName The name of the backend to be reset.
 */
async function resetBackend(backendName) {
  const ENGINE = tf.engine();
  if (!(backendName in ENGINE.registryFactory)) {
    throw new Error(`${backendName} backend is not registed.`);
  }

  const currentBackend = tf.getBackend();

  if (backendName in ENGINE.registry) {
    const backendFactory = tf.findBackendFactory(backendName);
    tf.removeBackend(backendName);
    tf.registerBackend(backendName, backendFactory);
  }

  if (currentBackend === backendName) {
    const isSuccessful = await tf.setBackend(backendName);
    if (!isSuccessful) {
      showMsg(`Failed to set backend ${backendName}.`);
      return false;
    }
  }

  return true;
}

/**
 * Get the renderer info from the WebGL backend.