Method getTensorInfos

tfjs-tflite/src/tflite_model_test.ts:72–122

(firstTensorType: TFLiteDataType = 'int32')

  }

  private getTensorInfos(firstTensorType: TFLiteDataType = 'int32'):
      TFLiteWebModelRunnerTensorInfo[] {
    const shape0 = [1, 2, 3];
    let buffer0: Int8Array|Uint8Array|Int16Array|Int32Array|Uint32Array|
        Float32Array|Float64Array = undefined;
    const size0 = shape0.reduce((a, c) => a * c, 1);
    switch (firstTensorType) {
      case 'int8':
        buffer0 = new Int8Array(size0);
        break;
      case 'uint8':
        buffer0 = new Uint8Array(size0);
        break;
      case 'int16':
        buffer0 = new Int16Array(size0);
        break;
      case 'int32':
        buffer0 = new Int32Array(size0);
        break;
      case 'uint32':
        buffer0 = new Uint32Array(size0);
        break;
      case 'float32':
        buffer0 = new Float32Array(size0);
        break;
      case 'float64':
        buffer0 = new Float64Array(size0);
        break;
      default:
        break;
    }

    const shape1 = [1, 2];
    const buffer1 = new Float32Array(shape1.reduce((a, c) => a * c, 1));
    return [
      {
        id: 0,
        dataType: firstTensorType,
        name: 't0',
        shape: shape0.join(','),
        data: () => buffer0,
      },
      {
        id: 1,
        dataType: 'float32',
        name: 't1',
        shape: shape1.join(','),
        data: () => buffer1,
      },
    ];
  }
}

let tfliteModel: TFLiteModel;