Function downloadValuesFromTensorContainer

e2e/benchmarks/benchmark_util.js:354–395

* Downloads the values from the `tensorContainer` from any `tf.Tensor`s found * within the `tensorContainer`. Returns a promise of `TypedArray` or * `TypedArray[]` that resolves when the computation has finished. * * The values are asynchronously downloaded in parallel. * * @param tensorContai

(tensorContainer)

 * @param tensorContainer The container of tensors to be downloaded.
 */
async function downloadValuesFromTensorContainer(tensorContainer) {
  let valueContainer;
  const readSync = tf.getBackend() === 'webgl';
  if (tensorContainer instanceof tf.Tensor) {
    if (readSync) {
      valueContainer = tensorContainer.dataSync();
    } else {
      valueContainer = await tensorContainer.data();
    }
  } else if (Array.isArray(tensorContainer)) {
    // Start value downloads from all tensors.
    const valuePromiseContainer = tensorContainer.map(async item => {
      if (item instanceof tf.Tensor) {
        if (readSync) {
          return item.dataSync();
        } else {
          return item.data();
        }
      }
      return item;
    });
    // Wait until all values are downloaded.
    valueContainer = await Promise.all(valuePromiseContainer);
  } else if (tensorContainer != null && typeof tensorContainer === 'object') {
    const valuePromiseContainer = [];
    // Start value downloads from all tensors.
    for (const property in tensorContainer) {
      if (tensorContainer[property] instanceof tf.Tensor) {
        if (readSync) {
          valuePromiseContainer.push(tensorContainer[property].dataSync());
        } else {
          valuePromiseContainer.push(tensorContainer[property].data());
        }
      } else {
        valuePromiseContainer.push(tensorContainer[property]);
      }
    }
    // Wait until all values are downloaded.
    valueContainer = await Promise.all(valuePromiseContainer);
  }
  return valueContainer;
}

/**
 * Executes the predict function for `model` (`model.predict` for