Function commonMobileNetPredictFunc

e2e/benchmarks/model_config.js:82–96

()

// Common predict function for MobileNetV3 and MobileNetV2Lite
function commonMobileNetPredictFunc() {
  const input = tf.randomNormal([1, 224, 224, 3]);
  const inputData = input.dataSync();
  if (typeof isTflite === 'function' && isTflite()) {
    return async () => {
      // Do copy from 'inputData' in each predict as its buffer
      // will be detached after transferring to worker.
      const input = inputData.slice(0);
      return await tfliteModel.predict(
          Comlink.transfer(input, [input.buffer]));
    };
  } else {
    return predictFunction(input);
  }
}

const benchmarks = {
  'MobileNetV3': {