Function loadWeightsAsArrayBuffer

tfjs-core/src/io/weights_loader.ts:38–72

(
  fetchURLs: string[], loadOptions?: LoadOptions)

 *   length as `fetchURLs`.
 */
export async function loadWeightsAsArrayBuffer(
  fetchURLs: string[], loadOptions?: LoadOptions): Promise<ArrayBuffer[]> {
  if (loadOptions == null) {
    loadOptions = {};
  }

  const fetchFunc = loadOptions.fetchFunc == null ? env().platform.fetch :
    loadOptions.fetchFunc;

  // Create the requests for all of the weights in parallel.
  const requests = fetchURLs.map(
    fetchURL =>
      fetchFunc(fetchURL, loadOptions.requestInit, { isBinary: true }));

  const fetchStartFraction = 0;
  const fetchEndFraction = 0.5;

  const responses = loadOptions.onProgress == null ?
    await Promise.all(requests) :
    await monitorPromisesProgress(
      requests, loadOptions.onProgress, fetchStartFraction,
      fetchEndFraction);

  const bufferPromises = responses.map(response => response.arrayBuffer());

  const bufferStartFraction = 0.5;
  const bufferEndFraction = 1;

  const buffers = loadOptions.onProgress == null ?
    await Promise.all(bufferPromises) :
    await monitorPromisesProgress(
      bufferPromises, loadOptions.onProgress, bufferStartFraction,
      bufferEndFraction);
  return buffers;
}

export function streamWeights(fetchURLs: string[], loadOptions: LoadOptions): ReadableStream<ArrayBuffer> {
  const fetchFunc = loadOptions.fetchFunc == null ? env().platform.fetch :